Example #1
0
 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)
Example #2
0
 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)
Example #3
0
 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)
Example #4
0
 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)
Example #5
0
 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)
Example #6
0
 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)