def test_read_csv_files(self): self.writefiles('.csv') inp = read_calculator_input(self.path) self.assertEqual(len(inp['fragility']), 5) self.assertEqual(len(inp['vulnerability']), 2) self.assertEqual(len(inp['exposure']), 37) self.assertEqual(len(inp['gmf']), 12)
def test_read_zip_file(self): self.writefiles('.csv') path = self.path + '.zip' zfile = zipfile.ZipFile(path, 'w') zfile.write(os.path.join(self.path, 'job.ini'), 'job.ini') for ftype in self.FILETYPES: zfile.write(os.path.join(self.path, ftype + '.csv'), ftype + '.csv') zfile.close() inp = read_calculator_input(path) self.assertEqual(len(inp['fragility']), 5) self.assertEqual(len(inp['vulnerability']), 2) self.assertEqual(len(inp['exposure']), 37) self.assertEqual(len(inp['gmf']), 12)