mirror of
https://github.com/Farama-Foundation/Gymnasium.git
synced 2025-08-23 15:04:20 +00:00
Fix get_wrapper_attr
/ set_wrapper_attr
. (#1293)
Co-authored-by: Mark Towers <mark.m.towers@gmail.com>
This commit is contained in:
@@ -215,6 +215,16 @@ def test_get_set_wrapper_attr():
|
||||
env.unwrapped._disable_render_order_enforcing
|
||||
assert env.get_wrapper_attr("_disable_render_order_enforcing") is True
|
||||
|
||||
# Test with top-most wrapper
|
||||
env.MY_ATTRIBUTE_1 = True
|
||||
assert env.get_wrapper_attr("MY_ATTRIBUTE_1") is True
|
||||
env.set_wrapper_attr("MY_ATTRIBUTE_1", False)
|
||||
assert env.get_wrapper_attr("MY_ATTRIBUTE_1") is False
|
||||
|
||||
# Test with non-existing attribute
|
||||
env.set_wrapper_attr("MY_ATTRIBUTE_2", True)
|
||||
assert getattr(env, "MY_ATTRIBUTE_2") is True
|
||||
|
||||
|
||||
class TestRandomSeeding:
|
||||
@staticmethod
|
||||
|
Reference in New Issue
Block a user