def test_get_coord_axes_bad_dims(self):
"""Test that ValueError is raised when
numpy array with incorrect dimensions
is fed to get_coord_axes()."""
with self.assertRaises(ValueError):
PSA.get_coord_axes(np.zeros((5, 5, 5, 5)))
def test_get_coord_axes_bad_dims(self):
"""Test that ValueError is raised when
numpy array with incorrect dimensions
is fed to get_coord_axes()."""
with self.assertRaises(ValueError):
PSA.get_coord_axes(np.zeros((5,5,5,5)))
def time_get_coord_axes(self, time_steps, n_atoms):
"""Benchmark get_coord_axes in psa module
"""
psa.get_coord_axes(path=self.path)