| 334 | self.assertTrue(mx.array_equal(z, mx.ones(64))) |
| 335 | |
| 336 | def test_comparisons(self): |
| 337 | a = mx.array([0.0, 1.0, 5.0]) |
| 338 | b = mx.array([-1.0, 2.0, 5.0]) |
| 339 | |
| 340 | self.assertEqual(mx.less(a, b).tolist(), [False, True, False]) |
| 341 | self.assertEqual(mx.less_equal(a, b).tolist(), [False, True, True]) |
| 342 | self.assertEqual(mx.greater(a, b).tolist(), [True, False, False]) |
| 343 | self.assertEqual(mx.greater_equal(a, b).tolist(), [True, False, True]) |
| 344 | |
| 345 | self.assertEqual(mx.less(a, 5).tolist(), [True, True, False]) |
| 346 | self.assertEqual(mx.less(5, a).tolist(), [False, False, False]) |
| 347 | self.assertEqual(mx.less_equal(5, a).tolist(), [False, False, True]) |
| 348 | self.assertEqual(mx.greater(a, 1).tolist(), [False, False, True]) |
| 349 | self.assertEqual(mx.greater_equal(a, 1).tolist(), [False, True, True]) |
| 350 | |
| 351 | a = mx.array([0.0, 1.0, 5.0, -1.0]) |
| 352 | b = mx.array([0.0, 2.0, 5.0, 3.0]) |
| 353 | self.assertEqual(mx.equal(a, b).tolist(), [True, False, True, False]) |
| 354 | self.assertEqual(mx.not_equal(a, b).tolist(), [False, True, False, True]) |
| 355 | |
| 356 | def test_array_equal(self): |
| 357 | x = mx.array([1, 2, 3, 4]) |