def test_linspace(): actual = onp.linspace(2.0, 3.0, dtype=onp.float32) expected = mnp.linspace(2.0, 3.0).asnumpy() match_array(actual, expected, error=6) actual = onp.linspace(2.0, 3.0, num=5, dtype=onp.float32) expected = mnp.linspace(2.0, 3.0, num=5).asnumpy() match_array(actual, expected, error=6) actual = onp.linspace( 2.0, 3.0, num=5, endpoint=False, dtype=onp.float32) expected = mnp.linspace(2.0, 3.0, num=5, endpoint=False).asnumpy() match_array(actual, expected, error=6) actual = onp.linspace(2.0, 3.0, num=5, retstep=True, dtype=onp.float32) expected = mnp.linspace(2.0, 3.0, num=5, retstep=True) match_array(actual[0], expected[0].asnumpy()) assert actual[1] == expected[1].asnumpy() actual = onp.linspace(2.0, [3, 4, 5], num=5, endpoint=False, dtype=onp.float32) expected = mnp.linspace( 2.0, [3, 4, 5], num=5, endpoint=False).asnumpy() match_array(actual, expected, error=6) actual = onp.linspace(2.0, [[3, 4, 5]], num=5, endpoint=False, axis=2) expected = mnp.linspace(2.0, [[3, 4, 5]], num=5, endpoint=False, axis=2).asnumpy() match_array(actual, expected, error=6) start = onp.random.random([2, 1, 4]).astype("float32") stop = onp.random.random([1, 5, 1]).astype("float32") actual = onp.linspace(start, stop, num=20, retstep=True, endpoint=False, dtype=onp.float32) expected = mnp.linspace(to_tensor(start), to_tensor(stop), num=20, retstep=True, endpoint=False) match_array(actual[0], expected[0].asnumpy(), error=6) match_array(actual[1], expected[1].asnumpy(), error=6) actual = onp.linspace(start, stop, num=20, retstep=True, endpoint=False, dtype=onp.int16) expected = mnp.linspace(to_tensor(start), to_tensor(stop), num=20, retstep=True, endpoint=False, dtype=mnp.int16) match_array(actual[0], expected[0].asnumpy(), error=6) match_array(actual[1], expected[1].asnumpy(), error=6) for axis in range(2): actual = onp.linspace(start, stop, num=20, retstep=False, endpoint=False, dtype=onp.float32, axis=axis) expected = mnp.linspace(to_tensor(start), to_tensor(stop), num=20, retstep=False, endpoint=False, dtype=mnp.float32, axis=axis) match_array(actual, expected.asnumpy(), error=6)
def test_linspace(): actual = onp.linspace(2.0, 3.0, dtype=onp.float32) expected = mnp.linspace(2.0, 3.0).asnumpy() match_array(actual, expected, error=7) actual = onp.linspace(2.0, 3.0, num=5, dtype=onp.float32) expected = mnp.linspace(2.0, 3.0, num=5).asnumpy() match_array(actual, expected, error=7) actual = onp.linspace(2.0, 3.0, num=5, endpoint=False, dtype=onp.float32) expected = mnp.linspace(2.0, 3.0, num=5, endpoint=False).asnumpy() match_array(actual, expected, error=7) actual = onp.linspace(2.0, 3.0, num=5, retstep=True, dtype=onp.float32) expected = mnp.linspace(2.0, 3.0, num=5, retstep=True) match_array(actual[0], expected[0].asnumpy()) assert actual[1] == expected[1] actual = onp.linspace(2.0, [3, 4, 5], num=5, endpoint=False, dtype=onp.float32) expected = mnp.linspace(2.0, [3, 4, 5], num=5, endpoint=False).asnumpy() match_array(actual, expected)
def test_linspace_exception(): with pytest.raises(TypeError): mnp.linspace(0, 1, num=2.5)
def linspace(start, stop, num, dtype=None): return mnp.linspace(start, stop, num, dtype=dtype)