Test #1 out of 2

Author: philipturner

Tests/PythonKitTests/PythonRuntimeTests.swift

@@ -327,10 +327,14 @@ class PythonRuntimeTests: XCTestCase {
         XCTAssertEqual(bytes, otherBytes)
     }
 
-    func testPythonFunction() {
+    private var canUsePythonFunction: Bool {
         let versionMajor = Python.versionInfo.major
         let versionMinor = Python.versionInfo.minor
-        guard (versionMajor == 3 && versionMinor >= 1) || versionMajor > 3 else {
+        return (versionMajor == 3 && versionMinor >= 1) || versionMajor > 3
+    }
+    
+    func testPythonFunction() {
+        guard canUsePythonFunction else {
             return
         }
 
@@ -344,4 +348,64 @@ class PythonRuntimeTests: XCTestCase {
         XCTAssertNotNil(Double(pythonSum))
         XCTAssertEqual(pythonSum, 5)
     }
+    
+    // From https://www.geeksforgeeks.org/create-classes-dynamically-in-python
+    func testPythonClassConstruction() {
+        guard canUsePythonFunction else {
+            return
+        }
+        
+        let constructor = PythonInstanceMethod { (params: [PythonObject]) in
+            let `self` = params[0]
+            let arg = params[1]
+            `self`.constructor_arg = arg
+            return Python.None
+        }
+
+        // Instead of calling `print`, use this to test what would be output.
+        var printOutput: String?
+
+        let displayMethod = PythonInstanceMethod { (params: [PythonObject]) in
+            // let `self` = params[0]
+            let arg = params[1]
+            printOutput = String(arg)
+            return Python.None
+        }
+
+        let classMethodOriginal = PythonInstanceMethod { (params: [PythonObject]) in
+            // let cls = params[0]
+            let arg = params[1]
+            printOutput = String(arg)
+            return Python.None
+        }
+
+        // Did not explicitly convert `constructor` or `displayMethod` to PythonObject.
+        // This is intentional, as the `PythonClass` initializer should take any
+        // `PythonConvertible` and not just `PythonObject`.
+        let classMethod = Python.classmethod(classMethodOriginal.pythonObject)
+
+        let Geeks = PythonClass("Geeks", members: [
+            // Constructor
+            "__init__": constructor,
+            
+            // Data members
+            "string_attribute": "Geeks 4 geeks!",
+            "int_attribute": 1706256,
+            
+            // Member functions
+            "func_arg": displayMethod,
+            "class_func": classMethod,
+        ]).pythonObject
+        
+        let obj = Geeks("constructor argument")
+        XCTAssertEqual(obj.constructor_arg, "constructor argument")
+        XCTAssertEqual(obj.string_attribute, "Geeks 4 geeks!")
+        XCTAssertEqual(obj.int_attribute, 1706256)
+
+        obj.func_arg("Geeks for Geeks")
+        XCTAssertEqual(printOutput, "Geeks for Geeks")
+
+        Geeks.class_func("Class Dynamically Created!")
+        XCTAssertEqual(printOutput, "Class Dynamically Created!")
+    }
 }