Example #1
0
    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)))
Example #2
0
    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)))
Example #3
0
 def time_get_coord_axes(self, time_steps, n_atoms):
     """Benchmark get_coord_axes in psa module
     """
     psa.get_coord_axes(path=self.path)