diff --git a/tests/nightly/test_large_vector.py b/tests/nightly/test_large_vector.py index 7ea27c221abf..3580bbd5ad48 100644 --- a/tests/nightly/test_large_vector.py +++ b/tests/nightly/test_large_vector.py @@ -56,17 +56,6 @@ def test_ndarray_ones(): assert nd.sum(a).asnumpy() == LARGE_X -@unittest.skip("need to fix") -def test_ndarray_convert(): - a = nd.zeros(shape=(LARGE_X, SMALL_Y)) - b = a.astype(np.int32) - b.wait_to_read() - assert b.dtype == np.int32 - b = a.tostype('row_sparse') - b.wait_to_read() - assert isinstance(b, mx.nd.sparse.RowSparseNDArray) - - @with_seed() def test_ndarray_random_uniform(): a = nd.random.uniform(shape=LARGE_X) @@ -107,14 +96,6 @@ def test_reduce(): assert nd.sum(a).asnumpy() == a.shape[0] * a.shape[1] -@unittest.skip("need to fix") -def test_dot(): - a = nd.ones(shape=(LARGE_X, SMALL_Y)) - b = nd.ones(shape=(SMALL_Y, SMALL_Y)) - res = nd.dot(a, b) - assert np.sum(res[-1].asnumpy() == SMALL_Y) == b.shape[1] - - def test_FullyConnected(): a = nd.ones(shape=(LARGE_X, SMALL_Y)) b = nd.ones(shape=(SMALL_Y, SMALL_Y)) @@ -200,17 +181,6 @@ def test_Dense(ctx=mx.cpu(0)): assert res.shape == (LARGE_X, 2) -@unittest.skip("need to fix") -def test_where(): - a = nd.ones(shape=(LARGE_X, SMALL_Y)) - b = nd.arange(0, LARGE_X * SMALL_Y).reshape(LARGE_X, SMALL_Y) - res = nd.where(b > 100, a, b) - assert np.sum(res[-1].asnumpy() == 1) == b.shape[1] - csr_cond = nd.sparse.cast_storage(b < 10, 'csr') - res = nd.sparse.where(csr_cond, a, b) - assert np.sum(res[0].asnumpy() == 1) == 10 - - def test_pick(): a = mx.nd.ones(shape=(LARGE_X, 2)) b = mx.nd.ones(shape=LARGE_X)