def test_read_1file_with_header(self): reader = CSVReader(self.file_with_header) self.assertEqual(reader.number_of_trajectories(), 1) self.assertEqual(reader.dimension(), self.nd) self.assertEqual(reader.n_frames_total(), self.nt) output = reader.get_output() np.testing.assert_almost_equal(output[0], self.data)
def test_with_multiple_files(self): files = [self.filename1, self.file_with_header] reader = CSVReader(files) self.assertEqual(reader.number_of_trajectories(), len(files))
def test_with_multiple_files(self): files = [self.filename1, self.filename1] reader = CSVReader(files) self.assertEqual(reader.number_of_trajectories(), len(files))