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)