From a19b21ad228141b24afb5e31b47fcae5952e89e8 Mon Sep 17 00:00:00 2001
From: Yury Popov <git@phoenix.dj>
Date: Sun, 20 Apr 2025 15:49:21 +0300
Subject: [PATCH] tests: add complex log1p

---
 python/tests/test_ops.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/python/tests/test_ops.py b/python/tests/test_ops.py
index 4fcb31f18..89bff378e 100644
--- a/python/tests/test_ops.py
+++ b/python/tests/test_ops.py
@@ -977,6 +977,13 @@ class TestOps(mlx_tests.MLXTestCase):
 
         self.assertTrue(np.allclose(result, expected))
 
+        # Complex test
+        a = mx.array([1, 0.5, 10, 100]) + 1j
+        result = mx.log1p(a)
+        expected = np.log1p(a, dtype=np.complex64)
+
+        self.assertTrue(np.allclose(result, expected))
+
     def test_sigmoid(self):
         a = mx.array([0.0, 1.0, -1.0, 5.0, -5.0])
         result = mx.sigmoid(a)