@@ -37,9 +37,7 @@ def __call__(self, context: NullContext) -> GenericResult:
3737 return GenericResult (value = "ok" )
3838
3939
40- # =============================================================================
4140# Tests for register_ccflow_import_path function
42- # =============================================================================
4341
4442
4543def test_base_module_available_after_import ():
@@ -117,9 +115,7 @@ def test_sync_to_module_registers_class_not_yet_on_module():
117115 assert getattr (module , unique_name , None ) is cls , "Class should be on module after sync"
118116
119117
120- # =============================================================================
121118# Tests for _register_on_module (internal function)
122- # =============================================================================
123119
124120
125121def test_register_on_module_uses_specified_module ():
@@ -168,9 +164,7 @@ def test_register_on_module_import_path_format():
168164 delattr (target_module , parts [1 ])
169165
170166
171- # =============================================================================
172167# Tests for local class registration via BaseModel
173- # =============================================================================
174168
175169
176170class TestLocalPersistencePreservesCloudpickle :
@@ -365,9 +359,7 @@ class LocalModel(BaseModel):
365359 assert restored_type_path == original_type_path
366360
367361
368- # =============================================================================
369362# Cross-process cloudpickle tests
370- # =============================================================================
371363
372364
373365@pytest .fixture
@@ -555,9 +547,7 @@ def test_cross_process_cloudpickle(self, pickle_file, create_code, load_code):
555547 assert "SUCCESS" in load_result .stdout
556548
557549
558- # =============================================================================
559550# Module-level classes should not be affected
560- # =============================================================================
561551
562552
563553class TestModuleLevelClassesUnaffected :
@@ -588,9 +578,7 @@ def test_module_level_standard_pickle_works(self):
588578 assert type (restored ) is ModuleLevelModel
589579
590580
591- # =============================================================================
592581# Ray task tests
593- # =============================================================================
594582
595583
596584class TestRayTaskWithLocalClasses :
@@ -690,9 +678,7 @@ def process_model(m):
690678 assert result .startswith ("hello|ccflow.local_persistence._Local_" )
691679
692680
693- # =============================================================================
694681# UUID uniqueness tests
695- # =============================================================================
696682
697683
698684class TestUUIDUniqueness :
@@ -747,9 +733,7 @@ class TestModel(BaseModel):
747733 assert all (c in "0123456789abcdef" for c in uuid_part )
748734
749735
750- # =============================================================================
751736# Nested class and inheritance tests
752- # =============================================================================
753737
754738
755739class OuterClass :
@@ -858,9 +842,7 @@ class LocalSubclass(ModuleLevelModel):
858842 assert "__ccflow_import_path__" not in ModuleLevelModel .__dict__
859843
860844
861- # =============================================================================
862845# Generic types tests
863- # =============================================================================
864846
865847
866848class TestGenericTypes :
@@ -915,9 +897,7 @@ class GenericModel(BaseModel, Generic[T]):
915897 assert GenericModel .__ccflow_import_path__ .startswith (local_persistence .LOCAL_ARTIFACTS_MODULE_NAME )
916898
917899
918- # =============================================================================
919900# Import string tests
920- # =============================================================================
921901
922902
923903class TestImportString :
@@ -955,9 +935,7 @@ def test_import_string_still_works_for_simple_paths(self):
955935 assert result is os .path .join
956936
957937
958- # =============================================================================
959938# Registration strategy tests
960- # =============================================================================
961939
962940
963941class TestRegistrationStrategy :
@@ -993,9 +971,7 @@ class LocalModel(BaseModel):
993971 assert "<locals>" in LocalModel .__qualname__
994972
995973
996- # =============================================================================
997974# Tests for create_ccflow_model wrapper
998- # =============================================================================
999975
1000976
1001977class TestCreateCcflowModelWrapper :
0 commit comments