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