Add a unit test for src.data_utils.get_data_types()

woctezuma · May 22, 2021 · ace5245 · ace5245
1 parent 6b4c7f4
commit ace5245
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/tests.py b/tests.py
@@ -16,6 +16,13 @@ def test_get_data_types(self):
         self.assertIn("items", data_types)
         self.assertIn("offers", data_types)
 
+    def test_get_data_file_name(self):
+        data_types = src.data_utils.get_data_types()
+        for data_type in data_types:
+            fname = src.data_utils.get_data_file_name(data_type)
+            expected_fname = f"data/{data_type}_list.json"
+            self.assertEqual(fname, expected_fname)
+
     def test_load_data(self):
         data = src.data_utils.load_data()
         self.assertGreater(len(data), 0)