def test_dump(self):
output_folder = tempfile.mkdtemp()
config_path = os.path.join("tests", "data", "cim", "data_generator", "dumps", "config.yml")
dump_from_config(config_path, output_folder, 20)
# check output folder
for fname in ["ports.csv", "vessels.csv", "routes.csv", "global_order_proportion.txt", "order_proportion.csv", "misc.yml"]:
self.assertTrue(os.path.exists(os.path.join(output_folder, fname)), fname)
def test_dump_with_invalid_parameters(self):
output_folder = tempfile.mkdtemp()
config_path = os.path.join("data", "cim", "data_generator", "dumps", "config.yml")
with self.assertRaises(AssertionError):
dump_from_config(None, output_folder, 20)
with self.assertRaises(AssertionError):
dump_from_config("", output_folder, 20)
with self.assertRaises(AssertionError):
dump_from_config(config_path, None, 20)