def test_zero_dimension(self): provider = WXFExprProvider(NumPyWXFEncoder()) arr = numpy.ndarray([2, 0, 3]) with self.assertRaises(Exception) as err: next(provider.provide_wxfexpr(arr)) self.assertEqual(force_text(err.exception), "Dimensions must be positive integers.")
def test_dimensions(self): provider = WXFExprProvider(NumPyWXFEncoder()) arr = numpy.ndarray([2, 1, 3]) wxfExpr = next(provider.provide_wxfexpr(arr)) self.assertEqual(wxfExpr.dimensions, (2, 1, 3))