def test_product_space_unflatten_n():
space = Product([Discrete(3), Discrete(3)])
np.testing.assert_array_equal(space.flatten((2, 2)), space.flatten_n([(2, 2)])[0])
np.testing.assert_array_equal(
space.unflatten(space.flatten((2, 2))),
space.unflatten_n(space.flatten_n([(2, 2)]))[0]
)
def action_space(self):
return Product([
Discrete(11),
Discrete(11),
Discrete(11),
Discrete(2),
Discrete(2)
])
def observation_space(self):
"""
Returns a Space object
"""
return Product(Discrete(52), # Player card 1
Discrete(52), # Player card 2
Discrete(52), # Player card 3
Discrete(52), # Dealer card 1
Discrete(52), # Dealer card 2
Discrete(52)) # Dealer card 3
def test_product_space():
_ = Product([Discrete(3), Discrete(2)])
product_space = Product(Discrete(3), Discrete(2))
sample = product_space.sample()
assert product_space.contains(sample)
