gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-08-19 13:32:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

redo black (#2272)

Browse Source
This commit is contained in:
J K Terry
2021-07-29 15:39:42 -04:00
committed by GitHub
parent ff8c269abb
commit 78d2b512d8
104 changed files with 1350 additions and 418 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
gym/vector/tests/test_numpy_utils.py
View File

@@ -10,7 +10,9 @@ from gym.vector.tests.utils import spaces
from gym.vector.utils.numpy_utils import concatenate, create_empty_array
@pytest.mark.parametrize("space", spaces, ids=[space.__class__.__name__ for space in spaces])
@pytest.mark.parametrize(
"space", spaces, ids=[space.__class__.__name__ for space in spaces]
)
def test_concatenate(space):
def assert_type(lhs, rhs, n):
# Special case: if rhs is a list of scalars, lhs must be an np.ndarray
@@ -51,7 +53,9 @@ def test_concatenate(space):
@pytest.mark.parametrize("n", [1, 8])
@pytest.mark.parametrize("space", spaces, ids=[space.__class__.__name__ for space in spaces])
@pytest.mark.parametrize(
"space", spaces, ids=[space.__class__.__name__ for space in spaces]
)
def test_create_empty_array(space, n):
def assert_nested_type(arr, space, n):
if isinstance(space, _BaseGymSpaces):
@@ -79,7 +83,9 @@ def test_create_empty_array(space, n):
@pytest.mark.parametrize("n", [1, 8])
@pytest.mark.parametrize("space", spaces, ids=[space.__class__.__name__ for space in spaces])
@pytest.mark.parametrize(
"space", spaces, ids=[space.__class__.__name__ for space in spaces]
)
def test_create_empty_array_zeros(space, n):
def assert_nested_type(arr, space, n):
if isinstance(space, _BaseGymSpaces):
@@ -107,7 +113,9 @@ def test_create_empty_array_zeros(space, n):
assert_nested_type(array, space, n=n)
@pytest.mark.parametrize("space", spaces, ids=[space.__class__.__name__ for space in spaces])
@pytest.mark.parametrize(
"space", spaces, ids=[space.__class__.__name__ for space in spaces]
)
def test_create_empty_array_none_shape_ones(space):
def assert_nested_type(arr, space):
if isinstance(space, _BaseGymSpaces):