diff --git a/tests/test_analytic.py b/tests/test_analytic.py
index 40b990b02db1625c0478ab5976efb695348f5055..ad78b4c9d92dba0516f769ddb6d5bb5680d58189 100644
--- a/tests/test_analytic.py
+++ b/tests/test_analytic.py
@@ -34,7 +34,7 @@ class TestCaseAnalytic(unittest.TestCase):
width = (self.dim[0]/2, self.dim[1]/2, self.dim[2]/2)
phase = an.phase_mag_slab(self.dim, self.res, self.phi, self.center, width)
reference = np.load(os.path.join(self.path, 'ref_phase_slab.npy'))
- np.testing.assert_equal(phase, reference, 'Unexpected behavior in phase_mag_slab()')
+ np.testing.assert_almost_equal(phase, reference, err_msg='Unexpected behavior in phase_mag_slab()')
def test_phase_mag_disc(self):
'''Test of the phase_mag_disc method.'''
@@ -42,14 +42,14 @@ class TestCaseAnalytic(unittest.TestCase):
height = self.dim[2]/2
phase = an.phase_mag_disc(self.dim, self.res, self.phi, self.center, radius, height)
reference = np.load(os.path.join(self.path, 'ref_phase_disc.npy'))
- np.testing.assert_equal(phase, reference, 'Unexpected behavior in phase_mag_disc()')
+ np.testing.assert_almost_equal(phase, reference, err_msg='Unexpected behavior in phase_mag_disc()')
def test_phase_mag_sphere(self):
'''Test of the phase_mag_sphere method.'''
radius = self.dim[2]/4
phase = an.phase_mag_sphere(self.dim, self.res, self.phi, self.center, radius)
reference = np.load(os.path.join(self.path, 'ref_phase_sphere.npy'))
- np.testing.assert_equal(phase, reference, 'Unexpected behavior in phase_mag_sphere()')
+ np.testing.assert_almost_equal(phase, reference, err_msg='Unexpected behavior in phase_mag_sphere()')
def test_phase_mag_vortex(self):
'''Test of the phase_mag_vortex method.'''
@@ -57,7 +57,7 @@ class TestCaseAnalytic(unittest.TestCase):
height = self.dim[2]/2
phase = an.phase_mag_vortex(self.dim, self.res, self.center, radius, height)
reference = np.load(os.path.join(self.path, 'ref_phase_vort.npy'))
- np.testing.assert_equal(phase, reference, 'Unexpected behavior in phase_mag_vortex()')
+ np.testing.assert_almost_equal(phase, reference, err_msg='Unexpected behavior in phase_mag_vortex()')
if __name__ == '__main__':
suite = unittest.TestLoader().loadTestsFromTestCase(TestCaseAnalytic)
diff --git a/tests/test_magcreator.py b/tests/test_magcreator.py
index 3a38ca7b53d0ac6ce3df22f532622e2896f500ae..32ad03b5a03026afa951dae350322401704905f1 100644
--- a/tests/test_magcreator.py
+++ b/tests/test_magcreator.py
@@ -20,52 +20,52 @@ class TestCaseMagCreator(unittest.TestCase):
def test_shape_slab(self):
test_slab = mc.Shapes.slab((5, 6, 7), (2, 3, 4), (1, 3, 5))
- np.testing.assert_equal(test_slab, np.load(os.path.join(self.path, 'ref_slab.npy')),
- 'Created slab does not match expectation!')
+ np.testing.assert_almost_equal(test_slab, np.load(os.path.join(self.path, 'ref_slab.npy')),
+ err_msg='Created slab does not match expectation!')
def test_shape_disc(self):
test_disc_z = mc.Shapes.disc((5, 6, 7), (2, 3, 4), 2, 3, 'z')
test_disc_y = mc.Shapes.disc((5, 6, 7), (2, 3, 4), 2, 3, 'y')
test_disc_x = mc.Shapes.disc((5, 6, 7), (2, 3, 4), 2, 3, 'x')
- np.testing.assert_equal(test_disc_z, np.load(os.path.join(self.path, 'ref_disc_z.npy')),
- 'Created disc in z-direction does not match expectation!')
- np.testing.assert_equal(test_disc_y, np.load(os.path.join(self.path, 'ref_disc_y.npy')),
- 'Created disc in y-direction does not match expectation!')
- np.testing.assert_equal(test_disc_x, np.load(os.path.join(self.path, 'ref_disc_x.npy')),
- 'Created disc in x-direction does not match expectation!')
+ np.testing.assert_almost_equal(test_disc_z, np.load(os.path.join(self.path, 'ref_disc_z.npy')),
+ err_msg='Created disc in z-direction does not match expectation!')
+ np.testing.assert_almost_equal(test_disc_y, np.load(os.path.join(self.path, 'ref_disc_y.npy')),
+ err_msg='Created disc in y-direction does not match expectation!')
+ np.testing.assert_almost_equal(test_disc_x, np.load(os.path.join(self.path, 'ref_disc_x.npy')),
+ err_msg='Created disc in x-direction does not match expectation!')
def test_shape_sphere(self):
test_sphere = mc.Shapes.sphere((5, 6, 7), (2, 3, 4), 2)
- np.testing.assert_equal(test_sphere, np.load(os.path.join(self.path, 'ref_sphere.npy')),
- 'Created sphere does not match expectation!')
+ np.testing.assert_almost_equal(test_sphere, np.load(os.path.join(self.path, 'ref_sphere.npy')),
+ err_msg='Created sphere does not match expectation!')
def test_shape_filament(self):
test_filament_z = mc.Shapes.filament((5, 6, 7), (2, 3), 'z')
test_filament_y = mc.Shapes.filament((5, 6, 7), (2, 3), 'y')
test_filament_x = mc.Shapes.filament((5, 6, 7), (2, 3), 'x')
- np.testing.assert_equal(test_filament_z, np.load(os.path.join(self.path, 'ref_fil_z.npy')),
- 'Created filament in z-direction does not match expectation!')
- np.testing.assert_equal(test_filament_y, np.load(os.path.join(self.path, 'ref_fil_y.npy')),
- 'Created filament in y-direction does not match expectation!')
- np.testing.assert_equal(test_filament_x, np.load(os.path.join(self.path, 'ref_fil_x.npy')),
- 'Created filament in x-direction does not match expectation!')
+ np.testing.assert_almost_equal(test_filament_z, np.load(os.path.join(self.path, 'ref_fil_z.npy')),
+ err_msg='Created filament in z-direction does not match expectation!')
+ np.testing.assert_almost_equal(test_filament_y, np.load(os.path.join(self.path, 'ref_fil_y.npy')),
+ err_msg='Created filament in y-direction does not match expectation!')
+ np.testing.assert_almost_equal(test_filament_x, np.load(os.path.join(self.path, 'ref_fil_x.npy')),
+ err_msg='Created filament in x-direction does not match expectation!')
def test_shape_pixel(self):
test_pixel = mc.Shapes.pixel((5, 6, 7), (2, 3, 4))
- np.testing.assert_equal(test_pixel, np.load(os.path.join(self.path, 'ref_pixel.npy')),
- 'Created pixel does not match expectation!')
+ np.testing.assert_almost_equal(test_pixel, np.load(os.path.join(self.path, 'ref_pixel.npy')),
+ err_msg='Created pixel does not match expectation!')
def test_create_mag_dist_homog(self):
mag_shape = mc.Shapes.disc((1, 10, 10), (0, 4.5, 4.5), 3, 1)
magnitude = mc.create_mag_dist_homog(mag_shape, pi/4)
- np.testing.assert_equal(magnitude, np.load(os.path.join(self.path, 'ref_mag_disc.npy')),
- 'Created homog. magnetic distribution does not match expectation')
+ np.testing.assert_almost_equal(magnitude, np.load(os.path.join(self.path, 'ref_mag_disc.npy')),
+ err_msg='Created homog. magnetic distribution does not match expectation')
def test_create_mag_dist_vortex(self):
mag_shape = mc.Shapes.disc((1, 10, 10), (0, 4.5, 4.5), 3, 1)
magnitude = mc.create_mag_dist_vortex(mag_shape)
- np.testing.assert_equal(magnitude, np.load(os.path.join(self.path, 'ref_mag_vort.npy')),
- 'Created vortex magnetic distribution does not match expectation')
+ np.testing.assert_almost_equal(magnitude, np.load(os.path.join(self.path, 'ref_mag_vort.npy')),
+ err_msg='Created vortex magnetic distribution does not match expectation')
if __name__ == '__main__':