Method test_long_column

python/tests/test_reduce.py:213–219

(self)

                    self.assertTrue(np.array_equal(out, ref, equal_nan=True))

    def test_long_column(self):
        a = (np.random.randn(8192, 64) * 32).astype(np.int32)
        b = mx.array(a)

        c1 = a.sum(0)
        c2 = b.sum(0)
        self.assertTrue(np.all(c1 == c2))


if __name__ == "__main__":