def test_only_npy(self):
reader = NumPyFileReader(self.npy_files)
from_files = [np.load(f) for f in self.npy_files]
concatenated = np.vstack(from_files)
output = reader.get_output()
self.assertEqual(reader.number_of_trajectories(), len(self.npy_files))
self.assertEqual(reader.n_frames_total(), concatenated.shape[0])
for x, y in zip(output, from_files):
np.testing.assert_array_almost_equal(x, y)
def test_only_npy(self):
reader = NumPyFileReader(self.npy_files)
from_files = [np.load(f) for f in self.npy_files]
concatenated = np.vstack(from_files)
output = reader.get_output()
self.assertEqual(reader.number_of_trajectories(), len(self.npy_files))
self.assertEqual(reader.n_frames_total(), concatenated.shape[0])
for x, y in zip(output, from_files):
np.testing.assert_array_almost_equal(x, y)
def testSingleFile(self):
reader = NumPyFileReader(self.npy_files[0])
self.assertEqual(reader.n_frames_total(), self.d.shape[0])
def testSingleFile(self):
reader = NumPyFileReader(self.npy_files[0])
self.assertEqual(reader.n_frames_total(), self.d.shape[0])
