Method test_integer_power

python/tests/test_ops.py:3210–3215

(self)

        self.assertTrue(mx.all(d == 1.0))

    def test_integer_power(self):
        x = mx.power(2, mx.array([8, 8, 8, 8, 8, 8, 8, 8]))
        self.assertTrue(mx.all(x == 256))

        # Doesn't hang
        x = mx.power(2, -1)

    def test_depends(self):
        a = mx.array([1.0, 2.0, 3.0])