Major fixes and new features

venv/lib/python3.12/site-packages/mypyc/test/test_emitclass.py

@@ -0,0 +1,35 @@
+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"