def testFromArray(self):
x = array([1, 2, 3])
self.assertEqual(x.shape, (3, ))
y = array([x, x])
self.assertEqual(y.shape, (2, 3))
z = array((x, x, x))
self.assertEqual(z.shape, (3, 3))
def testArray(self):
a = tensor([0, 1, 2], chunk_size=2)
b = array(a)
self.assertIsNot(a, b)
c = asarray(a)
self.assertIs(a, c)
def testDatatimeArith(self):
t1 = array([np.datetime64('2005-02-02'), np.datetime64('2005-02-03')])
t2 = t1 + np.timedelta64(1)
self.assertIsInstance(t2.op, AddConstant)
t3 = t1 - np.datetime64('2005-02-02')
self.assertIsInstance(t3.op, SubConstant)
self.assertEqual(t3.dtype,
(np.array(['2005-02-02', '2005-02-03'], dtype=np.datetime64) -
np.datetime64('2005-02-02')).dtype)
t1 = array([np.datetime64('2005-02-02'), np.datetime64('2005-02-03')])
subtract(t1, np.datetime64('2005-02-02'), out=empty(t1.shape, dtype=t3.dtype))
t1 = array([np.datetime64('2005-02-02'), np.datetime64('2005-02-03')])
add(t1, np.timedelta64(1, 'D'), out=t1)
def testDtypeFromOut(self):
x = array([-np.inf, 0., np.inf])
y = array([2, 2, 2])
t3 = isfinite(x, y)
self.assertEqual(t3.dtype, y.dtype)
