def test_blitz_discontinuous(self): aa2 = np.random.rand(2, 3) aa = aa2[:, 0] # Non-unit strides bb = copy.copy(aa) self.assertNotEqual(bb.strides[0], aa.strides[0]) ibmisc.example_double_blitz(aa) for a, b in zip(aa, bb): self.assertAlmostEqual(a, b * 2)
def test_blitz_discontinuous(self): aa2 = np.random.rand(2,3) aa = aa2[:,0] # Non-unit strides bb = copy.copy(aa) self.assertNotEqual(bb.strides[0], aa.strides[0]) ibmisc.example_double_blitz(aa) for a,b in zip(aa,bb): self.assertAlmostEqual(a,b*2)
def test_blitz_2d(self): aa = np.random.rand(2, 3) # Should only work on 1-D arrays with self.assertRaises(Exception): ibmisc.example_double_blitz(aa)
def test_blitz_1d(self): aa = np.array([float(i) for i in range(0, 5)]) bb = copy.copy(aa) ibmisc.example_double_blitz(aa) for a, b in zip(aa, bb): self.assertEqual(a, b * 2)
def test_blitz_2d(self): aa = np.random.rand(2,3) # Should only work on 1-D arrays with self.assertRaises(Exception): ibmisc.example_double_blitz(aa)
def test_blitz_1d(self): aa = np.array([float(i) for i in range(0,5)]) bb = copy.copy(aa) ibmisc.example_double_blitz(aa) for a,b in zip(aa,bb): self.assertEqual(a,b*2)