zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-10-31 16:21:27 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

real and imag properties (#2189)

Browse Source
This commit is contained in:
Awni Hannun
2025-05-15 18:17:50 -07:00
committed by GitHub
parent c1eb9d05d9
commit a2cadb8218
3 changed files with 23 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
python/tests/test_array.py
View File

@@ -2022,6 +2022,15 @@ class TestArray(mlx_tests.MLXTestCase):
with self.assertRaises(ValueError):
mx.add(y, x)
def test_real_imag(self):
x = mx.array([1.0])
self.assertEqual(x.real.item(), 1.0)
self.assertEqual(x.imag.item(), 0.0)
x = mx.array([1.0 + 1.0j])
self.assertEqual(x.imag.item(), 1.0)
self.assertEqual(x.real.item(), 1.0)
if __name__ == "__main__":
unittest.main()