rapydscript-ns 0.8.4 → 0.9.1

package/test/unit/language-service.js

@@ -1383,6 +1383,90 @@ function make_tests(Diagnostics, RS, STDLIB_MODULES) {
             },
         },
 
+        // ── Enum ──────────────────────────────────────────────────────────────
+
+        {
+            name: "enum_import_no_errors",
+            description: "from enum import Enum and class Color(Enum) produces no error markers",
+            run: function () {
+                var markers = d().check([
+                    "from enum import Enum",
+                    "class Color(Enum):",
+                    "    RED   = 1",
+                    "    GREEN = 2",
+                    "    BLUE  = 3",
+                    "x = Color.RED",
+                    "print(x.name)",
+                ].join("\n"));
+                var errors = markers.filter(function (m) { return m.severity === SEV_ERROR; });
+                assert.deepStrictEqual(errors, [],
+                    "Expected no errors for Enum usage but got: " + JSON.stringify(errors));
+            },
+        },
+
+        {
+            name: "enum_iteration_no_errors",
+            description: "for loop and list() over an Enum class produces no error markers",
+            run: function () {
+                var markers = d().check([
+                    "from enum import Enum",
+                    "class Status(Enum):",
+                    "    ACTIVE   = 1",
+                    "    INACTIVE = 0",
+                    "for m in Status:",
+                    "    print(m.name, m.value)",
+                    "members = list(Status)",
+                    "print(len(members))",
+                ].join("\n"));
+                var errors = markers.filter(function (m) { return m.severity === SEV_ERROR; });
+                assert.deepStrictEqual(errors, [],
+                    "Expected no errors for Enum iteration but got: " + JSON.stringify(errors));
+            },
+        },
+
+        {
+            name: "enum_isinstance_no_errors",
+            description: "isinstance(member, EnumClass) and isinstance(member, Enum) produce no errors",
+            run: function () {
+                var markers = d().check([
+                    "from enum import Enum",
+                    "class Color(Enum):",
+                    "    RED = 1",
+                    "ok = isinstance(Color.RED, Color)",
+                    "ok2 = isinstance(Color.RED, Enum)",
+                    "print(ok, ok2)",
+                ].join("\n"));
+                var errors = markers.filter(function (m) { return m.severity === SEV_ERROR; });
+                assert.deepStrictEqual(errors, [],
+                    "Expected no errors for isinstance/Enum but got: " + JSON.stringify(errors));
+            },
+        },
+
+        {
+            name: "format_dunder_no_errors",
+            description: "__format__ dunder in class body and format() / f-string usage produce no errors",
+            run: function () {
+                var markers = d().check([
+                    "class Money:",
+                    "    def __init__(self, amount):",
+                    "        self.amount = amount",
+                    "    def __str__(self):",
+                    "        return str(self.amount)",
+                    "    def __format__(self, spec):",
+                    "        if spec == 'usd':",
+                    "            return '$' + str(self.amount)",
+                    "        return format(self.amount, spec)",
+                    "m = Money(10)",
+                    "s1 = format(m, 'usd')",
+                    "s2 = str.format('{:usd}', m)",
+                    "s3 = f'{m:usd}'",
+                ].join("\n"));
+                var errors = markers.filter(function (m) { return m.severity === SEV_ERROR; });
+                assert.deepStrictEqual(errors, [],
+                    "Expected no errors for __format__ class but got: " + JSON.stringify(errors));
+            },
+        },
+
     ];
 
     return TESTS;