File tree Expand file tree Collapse file tree 1 file changed +7
-3
lines changed
Expand file tree Collapse file tree 1 file changed +7
-3
lines changed Original file line number Diff line number Diff line change 11import pytest
2+
23from groundlight .edge import (
34 DEFAULT ,
45 EDGE_ANSWERS_WITH_ESCALATION ,
89 InferenceConfig ,
910)
1011
12+ ONE_DETECTOR = 1
13+ TWO_DETECTORS = 2
14+
1115
1216def test_edge_endpoint_config_is_not_subclass_of_detectors_config ():
1317 assert not issubclass (EdgeEndpointConfig , DetectorsConfig )
@@ -32,7 +36,7 @@ def test_add_detector_allows_equivalent_named_inference_config():
3236 ),
3337 )
3438
35- assert len (detectors_config .detectors ) == 2
39+ assert len (detectors_config .detectors ) == TWO_DETECTORS
3640 assert list (detectors_config .edge_inference_configs .keys ()) == ["custom_config" ]
3741
3842
@@ -64,8 +68,8 @@ def test_from_detectors_config_copies_detector_data():
6468 config = EdgeEndpointConfig .from_detectors_config (detectors_config )
6569 detectors_config .add_detector ("det_2" , DEFAULT )
6670
67- assert len (config .detectors ) == 1
68- assert len (detectors_config .detectors ) == 2
71+ assert len (config .detectors ) == ONE_DETECTOR
72+ assert len (detectors_config .detectors ) == TWO_DETECTORS
6973
7074
7175def test_inference_config_validation_errors ():
You can’t perform that action at this time.
0 commit comments