mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
3603f646eb
Make `np.arange` respect an explicitly provided dtype. Also remove duplicated tests: - torch_np/test_function_base.py::TestArange is a dupe of - torch_np/numpy_tests/core/test_multiarray.py::TestArange Fixes #109975 Pull Request resolved: https://github.com/pytorch/pytorch/pull/110005 Approved by: https://github.com/lezcano
28 lines
797 B
Python
28 lines
797 B
Python
# Owner(s): ["module: dynamo"]
|
|
|
|
|
|
import pytest
|
|
|
|
import torch._numpy as np
|
|
from torch._numpy.testing import assert_equal
|
|
|
|
from torch.testing._internal.common_utils import run_tests, TestCase
|
|
|
|
|
|
class TestAppend(TestCase):
|
|
# tests taken from np.append docstring
|
|
def test_basic(self):
|
|
result = np.append([1, 2, 3], [[4, 5, 6], [7, 8, 9]])
|
|
assert_equal(result, np.arange(1, 10, dtype=int))
|
|
|
|
# When `axis` is specified, `values` must have the correct shape.
|
|
result = np.append([[1, 2, 3], [4, 5, 6]], [[7, 8, 9]], axis=0)
|
|
assert_equal(result, np.arange(1, 10, dtype=int).reshape((3, 3)))
|
|
|
|
with pytest.raises((RuntimeError, ValueError)):
|
|
np.append([[1, 2, 3], [4, 5, 6]], [7, 8, 9], axis=0)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
run_tests()
|