Method test_roll_errors

python/tests/test_ops.py:3093–3096

(self)

            self.assertTrue(mx.array_equal(y1, y2).item())

    def test_roll_errors(self):
        x = mx.array([])
        result = mx.roll(x, [0], [0])
        self.assertTrue(mx.array_equal(result, x))

    def test_real_imag(self):
        x = mx.random.uniform(shape=(4, 4))