zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-25 01:41:17 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

tests: add complex log1p

Browse Source
This commit is contained in:
Yury Popov 2025-04-20 15:49:21 +03:00
parent c3c2bdb194
commit a19b21ad22
No known key found for this signature in database
GPG Key ID: 76DE18AD6634F257
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
python/tests/test_ops.py
View File

@ -977,6 +977,13 @@ class TestOps(mlx_tests.MLXTestCase):
self.assertTrue(np.allclose(result, expected)) 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): def test_sigmoid(self):
a = mx.array([0.0, 1.0, -1.0, 5.0, -5.0]) a = mx.array([0.0, 1.0, -1.0, 5.0, -5.0])
result = mx.sigmoid(a) result = mx.sigmoid(a)