test: structure test scripts into tests directory and ignore artifacts

Author: solderzzc

.gitignore

@@ -14,3 +14,10 @@ DerivedData/
 .vscode/
 .idea/
 mlx-swift-lm
+
+# Temporary Artifacts & Logs
+*.log
+*.metallib
+*.pid
+curl_out.txt
+sample.txt

tests/measure_speed.py

@@ -0,0 +1,27 @@
+import time
+import urllib.request
+import json
+
+print("Starting inference request...", flush=True)
+start = time.time()
+try:
+    data = json.dumps({
+        "model": "mlx-community/Qwen3.5-122B-A10B-4bit",
+        "messages": [{"role": "user", "content": "Say hi"}],
+        "max_tokens": 5
+    }).encode('utf-8')
+    req = urllib.request.Request(
+        "http://127.0.0.1:5413/v1/chat/completions",
+        data=data,
+        headers={"Content-Type": "application/json"}
+    )
+    with urllib.request.urlopen(req, timeout=600) as response:
+        result = response.read().decode('utf-8')
+        duration = time.time() - start
+        print(f"Status Code: {response.status}")
+        print(f"Response: {result}")
+        print(f"Duration: {duration:.2f} seconds")
+except Exception as e:
+    duration = time.time() - start
+    print(f"Request failed: {e}")
+    print(f"Failed after {duration:.2f} seconds")

tests/test_assign.swift

@@ -0,0 +1 @@
+import MLX; var a = MLXArray.zeros([5, 2]); a[MLXArray([0, 2])] = MLXArray.ones([2, 2]); print(a)

tests/test_children.swift

@@ -0,0 +1,8 @@
+import Foundation
+
+func sumBytes(_ p: Progress) -> Int64 {
+    if p.children.isEmpty {
+        return p.completedUnitCount
+    }
+    return p.children.reduce(0) { $0 + sumBytes($1) }
+}

tests/test_inference.sh

@@ -0,0 +1,14 @@
+#!/bin/bash
+echo "=== Starting inference at $(date) ==="
+START=$(python3 -c 'import time; print(time.time())')
+curl -sf --max-time 600 -X POST http://127.0.0.1:5413/v1/chat/completions \
+  -H "Content-Type: application/json" \
+  -d '{"model":"mlx-community/Qwen3.5-122B-A10B-4bit","messages":[{"role":"user","content":"Say hi"}],"max_tokens":5}' > curl_out.txt
+RC=$?
+END=$(python3 -c 'import time; print(time.time())')
+DUR=$(python3 -c "print(f'{$END - $START:.2f}')")
+echo "=== Done at $(date) ==="
+echo "cURL Exit code: $RC"
+echo "Duration: $DUR seconds"
+cat curl_out.txt
+echo ""

tests/test_leaf.swift

@@ -0,0 +1 @@
+import MLXNN; class MLP: Module { @ModuleInfo(key: "proj") var proj = Linear(10, 10) }; let p = MLP(); print(p.leafModules().flattened().count)

tests/test_mirror.swift

@@ -0,0 +1,14 @@
+
+@propertyWrapper
+struct ModuleInfo<Value> {
+    var wrappedValue: Value
+    var key: String
+}
+class SwitchGLU {}
+class MLP { 
+  @ModuleInfo(key: "switch_mlp") var switchMLP: SwitchGLU = SwitchGLU()
+}
+let m = MLP()
+let mirror = Mirror(reflecting: m)
+for child in mirror.children { print(child.label!) }
+

tests/test_progress.swift

@@ -0,0 +1,5 @@
+import Foundation
+
+let p = Progress()
+print("throughput:", p.throughput ?? "nil")
+print("userInfo:", p.userInfo)

tests/test_speed.swift

@@ -0,0 +1,23 @@
+import Foundation
+
+func getSystemDownloadBytes() -> UInt64 {
+    var ifaddr: UnsafeMutablePointer<ifaddrs>?
+    guard getifaddrs(&ifaddr) == 0 else { return 0 }
+    defer { freeifaddrs(ifaddr) }
+    
+    var total: UInt64 = 0
+    var ptr = ifaddr
+    while let p = ptr {
+        if p.pointee.ifa_addr != nil, p.pointee.ifa_addr.pointee.sa_family == UInt8(AF_LINK) {
+            let data = p.pointee.ifa_data?.bindMemory(to: if_data.self, capacity: 1)
+            total += UInt64(data?.pointee.ifi_ibytes ?? 0)
+        }
+        ptr = p.pointee.ifa_next
+    }
+    return total
+}
+
+let start = getSystemDownloadBytes()
+Thread.sleep(forTimeInterval: 1.0)
+let end = getSystemDownloadBytes()
+print("Speed: \((Double(end - start) / 1048576.0)) MB/s")