Updated cleanup functionality

jmitchel3 · Jan 3, 2025 · 9ad5b80 · 9ad5b80
1 parent c31a625
commit 9ad5b80
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/discovery/test_fields.py b/tests/discovery/test_fields.py
@@ -16,7 +16,7 @@ def test_field_initialization(self):
         expected_choices = [
             (
                 "tests.discovery.tasks.custom_name_task",
-                "Custom Name Task (tests.discovery.tasks.custom_name_task)",
+                "Custom Name Task (tests.discovery.tasks)",
             ),
             (
                 "tests.discovery.tasks.debug_task",
@@ -39,7 +39,7 @@ def test_get_task(self):
         field = TaskChoiceField()
 
         # Set value using bound field data
-        field.data = "Custom Name Task (tests.discovery.tasks.custom_name_task)"
+        field.data = "Custom Name Task (tests.discovery.tasks)"
 
         # Should return the actual task path
         self.assertEqual(field.get_task(), "tests.discovery.tasks.custom_name_task")