Skip to content

Commit 22b8a07

Browse files
committed
fix tests
1 parent 41fdce9 commit 22b8a07

File tree

2 files changed

+10
-7
lines changed

2 files changed

+10
-7
lines changed

src/nglui/statebuilder/ngl_components.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1633,8 +1633,8 @@ def add_boxes(
16331633
self.add_annotations(
16341634
[
16351635
BoundingBoxAnnotation(
1636-
point_a=p_a,
1637-
point_b=p_b,
1636+
pointA=p_a,
1637+
pointB=p_b,
16381638
segments=seg,
16391639
description=d,
16401640
tags=t,

tests/test_base.py

Lines changed: 8 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -779,8 +779,8 @@ def test_datamap_priority_workflow(self):
779779
# Verify DataMaps are registered with correct priorities
780780
assert "high_priority_segments" in layer._datamaps
781781
assert "low_priority_segments" in layer._datamaps
782-
assert layer._datamaps["high_priority_segments"].priority == 1
783-
assert layer._datamaps["low_priority_segments"].priority == 10
782+
assert layer._datamap_priority["high_priority_segments"] == 1
783+
assert layer._datamap_priority["low_priority_segments"] == 10
784784

785785
# Layer should not be static anymore
786786
assert layer.is_static is False
@@ -856,10 +856,13 @@ def test_error_recovery_workflow(self):
856856
vs.add_segmentation_layer(source="precomputed://segments", name="segments")
857857
vs.add_segments([111, 222, 333])
858858

859-
# Should have 2 valid layers despite the error
860-
assert len(vs.layers) == 2
859+
# Should have 3 layers total (good_layer, bad_points created but failed, segments)
860+
assert len(vs.layers) == 3
861861
assert vs.layers[0].name == "good_layer"
862-
assert vs.layers[1].name == "segments"
862+
assert (
863+
vs.layers[1].name == "bad_points"
864+
) # Layer was created even though data failed
865+
assert vs.layers[2].name == "segments"
863866

864867
def test_method_chaining_workflow(self):
865868
"""Test that all methods support chaining properly"""

0 commit comments

Comments
 (0)