Extend tests

george0st · Aug 14, 2024 · 5ce06a7 · 5ce06a7
1 parent 91502bb
commit 5ce06a7
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/test_generator.py b/tests/test_generator.py
@@ -104,6 +104,17 @@ def test_generate_super_smallbulk(self):
         self.assertTrue(os.path.exists(path.join(dir, f"{basic_party.BasicParty.NAME}.csv")))
         self.assertTrue(os.path.exists(path.join(dir, f"{basic_contact.BasicContact.NAME}.csv")))
 
+    def test_generate_bigbigbulk(self):
+        lbl = "0-size-7000,2000"
+
+        generator = SyntheticData(os.path.join("..","01-model"),TestGenerator.OUTPUT_ADR, TestGenerator.OUTPUT_ADR)
+        generator.generate(label=lbl, count=7000, bulk_max=2000, compress=False)
+
+        dir = path.join(TestGenerator.OUTPUT_ADR, lbl)
+        self.assertTrue(os.path.exists(dir))
+        self.assertTrue(os.path.exists(path.join(dir, f"{basic_party.BasicParty.NAME}.csv")))
+        self.assertTrue(os.path.exists(path.join(dir, f"{basic_contact.BasicContact.NAME}.csv")))
+
     def test_generate_bigbulk(self):
         lbl = "0-size-2000,2000"