mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
7abeb92796
- Add `if __name__ == "__main__": run_tests()` stanzas to test files in `torch_np` folder so that these tests run on CI - Skip / xfail things smoked out by this change - remove a stray python file which should not have been added to tests in the first place. - fix einsum if opt_einsum is present - add skips for older numpies Pull Request resolved: https://github.com/pytorch/pytorch/pull/108762 Approved by: https://github.com/lezcano
57 lines
1.2 KiB
Python
57 lines
1.2 KiB
Python
# Owner(s): ["module: dynamo"]
|
|
|
|
"""Light smoke test switching between numpy to pytorch random streams.
|
|
"""
|
|
import pytest
|
|
|
|
import torch._numpy as tnp
|
|
from torch._numpy.testing import assert_equal
|
|
|
|
|
|
def test_uniform():
|
|
r = tnp.random.uniform(0, 1, size=10)
|
|
|
|
|
|
def test_shuffle():
|
|
x = tnp.arange(10)
|
|
tnp.random.shuffle(x)
|
|
|
|
|
|
def test_numpy_global():
|
|
tnp.random.USE_NUMPY_RANDOM = True
|
|
tnp.random.seed(12345)
|
|
x = tnp.random.uniform(0, 1, size=11)
|
|
|
|
# check that the stream is identical to numpy's
|
|
import numpy as _np
|
|
|
|
_np.random.seed(12345)
|
|
x_np = _np.random.uniform(0, 1, size=11)
|
|
|
|
assert_equal(x, tnp.asarray(x_np))
|
|
|
|
# switch to the pytorch stream, variates differ
|
|
tnp.random.USE_NUMPY_RANDOM = False
|
|
tnp.random.seed(12345)
|
|
|
|
x_1 = tnp.random.uniform(0, 1, size=11)
|
|
assert not (x_1 == x).all()
|
|
|
|
|
|
def test_wrong_global():
|
|
try:
|
|
oldstate = tnp.random.USE_NUMPY_RANDOM
|
|
|
|
tnp.random.USE_NUMPY_RANDOM = "oops"
|
|
with pytest.raises(ValueError):
|
|
tnp.random.rand()
|
|
|
|
finally:
|
|
tnp.random.USE_NUMPY_RANDOM = oldstate
|
|
|
|
|
|
if __name__ == "__main__":
|
|
from torch._dynamo.test_case import run_tests
|
|
|
|
run_tests()
|