def test_empty_arg(self):
from numba import typesystem as nt
empty_t = nt.module_attribute(module=np, attr='empty')
zeros_t = nt.module_attribute(module=np, attr='zeros')
ones_t = nt.module_attribute(module=np, attr='ones')
sig, symtab = infer(_empty_arg, functype(None, [int_, empty_t,
zeros_t, ones_t]))
for i in range(1, 4):
self.assertEqual(symtab['a%d' % i].type, double[:])
def module_attribute_type(obj):
"""
See if the object is registered to any module which might handle
type inference on the object.
"""
result = module_registry.lookup_module_attribute(obj)
if result is not None:
module, attr = result
return typesystem.module_attribute(module=module, attr=attr)
return None
def module_attribute_type(obj):
"""
See if the object is registered to any module which might handle
type inference on the object.
"""
result = module_registry.lookup_module_attribute(obj)
if result is not None:
module, attr = result
return typesystem.module_attribute(module=module, attr=attr)
return None
