| 517 | self.assertEqual(y.item(), 3.0 + 0j) |
| 518 | |
| 519 | def test_array_repr(self): |
| 520 | x = mx.array(True) |
| 521 | self.assertEqual(str(x), "array(True, dtype=bool)") |
| 522 | x = mx.array(1) |
| 523 | self.assertEqual(str(x), "array(1, dtype=int32)") |
| 524 | x = mx.array(1.0) |
| 525 | self.assertEqual(str(x), "array(1, dtype=float32)") |
| 526 | |
| 527 | x = mx.array([1, 0, 1]) |
| 528 | self.assertEqual(str(x), "array([1, 0, 1], dtype=int32)") |
| 529 | |
| 530 | x = mx.array([1] * 6) |
| 531 | expected = "array([1, 1, 1, 1, 1, 1], dtype=int32)" |
| 532 | self.assertEqual(str(x), expected) |
| 533 | |
| 534 | x = mx.array([1] * 7) |
| 535 | expected = "array([1, 1, 1, ..., 1, 1, 1], dtype=int32)" |
| 536 | self.assertEqual(str(x), expected) |
| 537 | |
| 538 | x = mx.array([[1, 2], [1, 2], [1, 2]]) |
| 539 | expected = "array([[1, 2],\n [1, 2],\n [1, 2]], dtype=int32)" |
| 540 | self.assertEqual(str(x), expected) |
| 541 | |
| 542 | x = mx.array([[[1, 2], [1, 2]], [[1, 2], [1, 2]]]) |
| 543 | expected = ( |
| 544 | "array([[[1, 2],\n" |
| 545 | " [1, 2]],\n" |
| 546 | " [[1, 2],\n" |
| 547 | " [1, 2]]], dtype=int32)" |
| 548 | ) |
| 549 | self.assertEqual(str(x), expected) |
| 550 | |
| 551 | x = mx.array([[1, 2]] * 6) |
| 552 | expected = ( |
| 553 | "array([[1, 2],\n" |
| 554 | " [1, 2],\n" |
| 555 | " [1, 2],\n" |
| 556 | " [1, 2],\n" |
| 557 | " [1, 2],\n" |
| 558 | " [1, 2]], dtype=int32)" |
| 559 | ) |
| 560 | self.assertEqual(str(x), expected) |
| 561 | x = mx.array([[1, 2]] * 7) |
| 562 | expected = ( |
| 563 | "array([[1, 2],\n" |
| 564 | " [1, 2],\n" |
| 565 | " [1, 2],\n" |
| 566 | " ...,\n" |
| 567 | " [1, 2],\n" |
| 568 | " [1, 2],\n" |
| 569 | " [1, 2]], dtype=int32)" |
| 570 | ) |
| 571 | self.assertEqual(str(x), expected) |
| 572 | |
| 573 | x = mx.array([1], dtype=mx.int8) |
| 574 | expected = "array([1], dtype=int8)" |
| 575 | self.assertEqual(str(x), expected) |
| 576 | x = mx.array([1], dtype=mx.int16) |