fix test

duneanalytics · Nov 15, 2022 · 6298bf3 · 6298bf3
1 parent 14326e8
commit 6298bf3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/tests/unit/dev/test.json.csv b/tests/unit/dev/test.json.csv
diff --git a/tests/unit/test_file.py b/tests/unit/test_file.py
@@ -51,6 +51,9 @@ def test_skip_empty_write(self):
                 file_manager.load(TEST_FILE)
 
     def test_file_type(self):
-        self.assertEqual(FileType.CSV, FileType.from_str("csv"))
-        self.assertEqual(FileType.JSON, FileType.from_str("json"))
-        self.assertEqual(FileType.JSON, FileType.from_str("ndjson"))
+        for enum_instance in FileType:
+            self.assertEqual(enum_instance, FileType.from_str(str(enum_instance)))
+        # The above is equivalent to, but also covers new items when added.
+        # self.assertEqual(FileType.CSV, FileType.from_str("csv"))
+        # self.assertEqual(FileType.JSON, FileType.from_str("json"))
+        # self.assertEqual(FileType.NDJSON, FileType.from_str("ndjson"))