def test_import_delays_B(self): """ Single delay """ de, n_de = IOM.import_delays(self.file_dict_B, self.fileformat, flag_verbose = self.flag_verbose) self.assertEqual(n_de, 1) self.assertEqual(de[0], 300) self.assertEqual(de[-1], 300)
def test_import_delays_A(self): """ Multiple delays """ de, n_de = IOM.import_delays(self.file_dict_A, self.fileformat, flag_verbose = self.flag_verbose) self.assertEqual(n_de, 5) self.assertEqual(de[0], 0) self.assertEqual(de[-1], 400)