Tipragot
628be439b8
Cela permet de ne pas avoir de problèmes de compatibilité car python est dans le git.
36 lines
1.2 KiB
Python
36 lines
1.2 KiB
Python
from __future__ import annotations
|
|
|
|
import unittest
|
|
|
|
from mypyc.codegen.emitclass import getter_name, setter_name, slot_key
|
|
from mypyc.ir.class_ir import ClassIR
|
|
from mypyc.namegen import NameGenerator
|
|
|
|
|
|
class TestEmitClass(unittest.TestCase):
|
|
def test_slot_key(self) -> None:
|
|
attrs = ["__add__", "__radd__", "__rshift__", "__rrshift__", "__setitem__", "__delitem__"]
|
|
s = sorted(attrs, key=lambda x: slot_key(x))
|
|
# __delitem__ and reverse methods should come last.
|
|
assert s == [
|
|
"__add__",
|
|
"__rshift__",
|
|
"__setitem__",
|
|
"__delitem__",
|
|
"__radd__",
|
|
"__rrshift__",
|
|
]
|
|
|
|
def test_setter_name(self) -> None:
|
|
cls = ClassIR(module_name="testing", name="SomeClass")
|
|
generator = NameGenerator([["mod"]])
|
|
|
|
# This should never be `setup`, as it will conflict with the class `setup`
|
|
assert setter_name(cls, "up", generator) == "testing___SomeClass_set_up"
|
|
|
|
def test_getter_name(self) -> None:
|
|
cls = ClassIR(module_name="testing", name="SomeClass")
|
|
generator = NameGenerator([["mod"]])
|
|
|
|
assert getter_name(cls, "down", generator) == "testing___SomeClass_get_down"
|