from pypy.module.micronumpy.test.test_base import BaseNumpyAppTest from pypy.conftest import option class AppTestObjectDtypes(BaseNumpyAppTest): spaceconfig = dict(usemodules=["micronumpy", "struct", "binascii"]) def setup_class(cls): BaseNumpyAppTest.setup_class.im_func(cls) cls.w_runappdirect = cls.space.wrap(option.runappdirect) def test_scalar_from_object(self): from numpy import array import sys class Polynomial(object): def whatami(self): return 'an object' a = array(Polynomial()) assert a.shape == () assert a.sum().whatami() == 'an object' def test_uninitialized_object_array_is_filled_by_None(self): import numpy as np a = np.ndarray([5], dtype="O") assert a[0] == None def test_object_arrays_add(self): import numpy as np a = np.array(["foo"], dtype=object) b = np.array(["bar"], dtype=object) raises(TypeError, np.add, a, 1) res = a + b assert res[0] == "foobar" def test_bool_func(self): import numpy as np a = np.array(["foo"], dtype=object) b = a and complex(1, -1) assert b == complex(1, -1) b = np.array(complex(1, -1)) and a assert (b == a).all() c = np.array([1, 2, 3]) assert (a[0] != c[0]) assert (c[0] != a[0]) assert (a[0] > c[0]) assert (not a[0] < c[0]) assert (c[0] < a[0]) assert (not c[0] > a[0]) def test_logical_ufunc(self): import numpy as np import sys a = np.array(["foo"], dtype=object) b = np.array([1], dtype=object) d = np.array([complex(1, 10)], dtype=object) c = np.logical_and(a, 1) assert c.dtype == np.dtype('object') assert c == 1 c = np.logical_and(b, complex(1, -1)) assert c.dtype == np.dtype('object') assert c == complex(1, -1) c = np.logical_and(d, b) assert c == 1 c = b & 1 assert c.dtype == np.dtype('object') assert (c == 1).all() c = np.array(1) & b assert (c == b).all() def test_reduce(self): import numpy as np class O(object): def whatami(self): return 'an object' fiveOs = [O()] * 5 a = np.array(fiveOs, dtype=object) print np.maximum b = np.maximum.reduce(a) assert b is not None def test_complex_op(self): import numpy as np import sys a = np.array(['abc', 'def'], dtype=object) b = np.array([1, 2, 3], dtype=object) c = np.array([complex(1, 1), complex(1, -1)], dtype=object) for arg in (a,b,c): assert (arg == np.real(arg)).all() assert (0 == np.imag(arg)).all() if '__pypy__' in sys.builtin_module_names: skip('not implemented yet') raises(AttributeError, np.conj, a) res = np.conj(b) assert (res == b).all() res = np.conj(c) assert res[0] == c[1] and res[1] == c[0] def test_keep_object_alive(self): # only translated does it really test the gc import numpy as np import gc class O(object): def whatami(self): return 'an object' fiveOs = [O()] * 5 a = np.array(fiveOs, dtype=object) del fiveOs gc.collect() assert a[2].whatami() == 'an object' def test_array_interface(self): import numpy as np class DummyArray(object): def __init__(self, interface, base=None): self.__array_interface__ = interface self.base = base a = np.array([(1, 2, 3)], dtype='u4,u4,u4') b = np.array([(1, 2, 3), (4, 5, 6), (7, 8, 9)], dtype='u4,u4,u4') interface = dict(a.__array_interface__) interface['shape'] = tuple([3]) interface['strides'] = tuple([0]) c = np.array(DummyArray(interface, base=a)) c.dtype = a.dtype #print c assert (c == np.array([(1, 2, 3), (1, 2, 3), (1, 2, 3)], dtype='u4,u4,u4') ).all() def test_for_object_scalar_creation(self): import numpy as np import sys a = np.object_() b = np.object_(3) b2 = np.object_(3.0) c = np.object_([4, 5]) d = np.array([None])[0] assert a is None assert type(b) is int assert type(b2) is float assert type(c) is np.ndarray assert c.dtype == object assert type(d) is type(None) if '__pypy__' in sys.builtin_module_names: skip('not implemented yet') e = np.object_([None, {}, []]) assert e.dtype == object def test_mem_array_creation_invalid_specification(self): # while not specifically testing object dtype, this # test segfaulted during ObjectType.store due to # missing gc hooks import numpy as np import sys ytype = np.object_ if '__pypy__' in sys.builtin_module_names: dt = np.dtype([('x', int), ('y', ytype)]) x = np.empty((4, 0), dtype = dt) raises(NotImplementedError, x.__getitem__, 'y') ytype = str dt = np.dtype([('x', int), ('y', ytype)]) # Correct way a = np.array([(1, 'object')], dt) # Wrong way - should complain about writing buffer to object dtype raises(ValueError, np.array, [1, 'object'], dt) def test_astype(self): import numpy as np a = np.array([b'a' * 100], dtype='O') assert 'a' * 100 in str(a) b = a.astype('S') assert b.dtype == 'S100' assert 'a' * 100 in str(b) a = np.array([u'a' * 100], dtype='O') assert 'a' * 100 in str(a) b = a.astype('U') assert b.dtype == 'U100' assert 'a' * 100 in str(b) a = np.array([123], dtype='U') assert a[0] == u'123' b = a.astype('O') assert b[0] == u'123' assert type(b[0]) is unicode class MyFloat(object): def __float__(self): return 1.0 a = np.array([MyFloat()]) assert a.shape == (1,) assert a.dtype == np.object_ b = a.astype(float) assert b.shape == (1,) assert b.dtype == np.float_ assert (b == 1.0).all() def test__reduce__(self): from numpy import arange, dtype from cPickle import loads, dumps import sys a = arange(15).astype(object) if '__pypy__' in sys.builtin_module_names: raises(NotImplementedError, dumps, a) skip('not implemented yet') b = loads(dumps(a)) assert (a == b).all() a = arange(15).astype(object).reshape((3, 5)) b = loads(dumps(a)) assert (a == b).all()