diff --git a/mypyc/primitives/float_ops.py b/mypyc/primitives/float_ops.py index ad028a901222..3359cf6fe122 100644 --- a/mypyc/primitives/float_ops.py +++ b/mypyc/primitives/float_ops.py @@ -2,12 +2,18 @@ from mypyc.ir.ops import ERR_MAGIC from mypyc.ir.rtypes import ( - str_rprimitive, float_rprimitive + str_rprimitive, float_rprimitive, object_rprimitive ) from mypyc.primitives.registry import ( - function_op + load_address_op, function_op ) +# Get the 'builtins.float' type object. +load_address_op( + name='builtins.float', + type=object_rprimitive, + src='PyFloat_Type') + # float(str) function_op( name='builtins.float', diff --git a/mypyc/primitives/tuple_ops.py b/mypyc/primitives/tuple_ops.py index ce88a4ee0f4d..33f8e331b56d 100644 --- a/mypyc/primitives/tuple_ops.py +++ b/mypyc/primitives/tuple_ops.py @@ -9,8 +9,13 @@ tuple_rprimitive, int_rprimitive, list_rprimitive, object_rprimitive, c_pyssize_t_rprimitive, bit_rprimitive ) -from mypyc.primitives.registry import method_op, function_op, custom_op +from mypyc.primitives.registry import load_address_op, method_op, function_op, custom_op +# Get the 'builtins.tuple' type object. +load_address_op( + name='builtins.tuple', + type=object_rprimitive, + src='PyTuple_Type') # tuple[index] (for an int index) tuple_get_item_op = method_op( diff --git a/mypyc/test-data/irbuild-dunders.test b/mypyc/test-data/irbuild-dunders.test index 808617e43889..d06a570aa7b0 100644 --- a/mypyc/test-data/irbuild-dunders.test +++ b/mypyc/test-data/irbuild-dunders.test @@ -175,16 +175,13 @@ L0: def f(c): c :: __main__.C r0, r1 :: int - r2, r3, r4 :: object - r5 :: str - r6, r7 :: object + r2, r3, r4, r5 :: object L0: r0 = c.__neg__() r1 = c.__invert__() r2 = load_address PyLong_Type r3 = PyObject_CallFunctionObjArgs(r2, c, 0) - r4 = builtins :: module - r5 = 'float' - r6 = CPyObject_GetAttr(r4, r5) - r7 = PyObject_CallFunctionObjArgs(r6, c, 0) + r4 = load_address PyFloat_Type + r5 = PyObject_CallFunctionObjArgs(r4, c, 0) return 1 + diff --git a/mypyc/test-data/run-python37.test b/mypyc/test-data/run-python37.test index 5bf2c29263e1..61d428c17a44 100644 --- a/mypyc/test-data/run-python37.test +++ b/mypyc/test-data/run-python37.test @@ -70,6 +70,7 @@ class Person4: @dataclass class Person5: + weight: float friends: Set[str] = field(default_factory=set) parents: FrozenSet[str] = frozenset() @@ -122,7 +123,8 @@ assert i8 > i9 assert Person1.__annotations__ == {'age': int, 'name': str} assert Person2.__annotations__ == {'age': int, 'name': str} -assert Person5.__annotations__ == {'friends': set, 'parents': frozenset} +assert Person5.__annotations__ == {'weight': float, 'friends': set, + 'parents': frozenset} [file driver.py] import sys diff --git a/mypyc/test-data/run-tuples.test b/mypyc/test-data/run-tuples.test index 759177342fa9..26b039320844 100644 --- a/mypyc/test-data/run-tuples.test +++ b/mypyc/test-data/run-tuples.test @@ -95,6 +95,49 @@ class Sub(NT): pass assert f(Sub(3, 2)) == 3 +-- Ref: https://github.com/mypyc/mypyc/issues/924 +[case testNamedTupleClassSyntax] +from typing import Dict, List, NamedTuple, Optional, Tuple, Union + +class ClassIR: pass + +class FuncIR: pass + +StealsDescription = Union[bool, List[bool]] + +class Record(NamedTuple): + st_mtime: float + st_size: int + is_borrowed: bool + hash: str + python_path: Tuple[str, ...] + type: 'ClassIR' + method: FuncIR + shadow_method: Optional[FuncIR] + classes: Dict[str, 'ClassIR'] + steals: StealsDescription + ordering: Optional[List[int]] + extra_int_constants: List[Tuple[int]] + +[file driver.py] +from typing import Optional +from native import ClassIR, FuncIR, Record + +assert Record.__annotations__ == { + 'st_mtime': float, + 'st_size': int, + 'is_borrowed': bool, + 'hash': str, + 'python_path': tuple, + 'type': ClassIR, + 'method': FuncIR, + 'shadow_method': type, + 'classes': dict, + 'steals': type, + 'ordering': type, + 'extra_int_constants': list, +}, Record.__annotations__ + [case testTupleOps] from typing import Tuple, List, Any, Optional from typing_extensions import Final