Build and test python package on Windows CI (#3049)

2026-01-24 07:22:36 +09:00
parent dc81c1503a
commit fd27829efa
10 changed files with 44 additions and 12 deletions
@@ -18,8 +18,8 @@ class TestMemory(mlx_tests.MLXTestCase):
        self.assertEqual(mx.set_cache_limit(old_limit), old_limit)

        old_limit = mx.set_memory_limit(10)
-        self.assertTrue(mx.set_memory_limit(old_limit), 10)
-        self.assertTrue(mx.set_memory_limit(old_limit), old_limit)
+        self.assertEqual(mx.set_memory_limit(old_limit), 10)
+        self.assertEqual(mx.set_memory_limit(old_limit), old_limit)

        # Query active and peak memory
        a = mx.zeros((4096,))