mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-24 10:16:09 +00:00
adding legacy to existing tests
This commit is contained in:
@@ -60,6 +60,7 @@ def test_strategy_config_types():
|
|||||||
BaseStrategyConfig,
|
BaseStrategyConfig,
|
||||||
DAggerStrategyConfig,
|
DAggerStrategyConfig,
|
||||||
HighlightStrategyConfig,
|
HighlightStrategyConfig,
|
||||||
|
LegacyStrategyConfig,
|
||||||
SentryStrategyConfig,
|
SentryStrategyConfig,
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -67,6 +68,7 @@ def test_strategy_config_types():
|
|||||||
assert SentryStrategyConfig().type == "sentry"
|
assert SentryStrategyConfig().type == "sentry"
|
||||||
assert HighlightStrategyConfig().type == "highlight"
|
assert HighlightStrategyConfig().type == "highlight"
|
||||||
assert DAggerStrategyConfig().type == "dagger"
|
assert DAggerStrategyConfig().type == "dagger"
|
||||||
|
assert LegacyStrategyConfig().type == "legacy"
|
||||||
|
|
||||||
|
|
||||||
def test_dagger_config_invalid_input_device():
|
def test_dagger_config_invalid_input_device():
|
||||||
@@ -203,6 +205,8 @@ def test_create_strategy_dispatches():
|
|||||||
BaseStrategyConfig,
|
BaseStrategyConfig,
|
||||||
DAggerStrategy,
|
DAggerStrategy,
|
||||||
DAggerStrategyConfig,
|
DAggerStrategyConfig,
|
||||||
|
LegacyStrategy,
|
||||||
|
LegacyStrategyConfig,
|
||||||
SentryStrategy,
|
SentryStrategy,
|
||||||
SentryStrategyConfig,
|
SentryStrategyConfig,
|
||||||
create_strategy,
|
create_strategy,
|
||||||
@@ -211,6 +215,7 @@ def test_create_strategy_dispatches():
|
|||||||
assert isinstance(create_strategy(BaseStrategyConfig()), BaseStrategy)
|
assert isinstance(create_strategy(BaseStrategyConfig()), BaseStrategy)
|
||||||
assert isinstance(create_strategy(SentryStrategyConfig()), SentryStrategy)
|
assert isinstance(create_strategy(SentryStrategyConfig()), SentryStrategy)
|
||||||
assert isinstance(create_strategy(DAggerStrategyConfig()), DAggerStrategy)
|
assert isinstance(create_strategy(DAggerStrategyConfig()), DAggerStrategy)
|
||||||
|
assert isinstance(create_strategy(LegacyStrategyConfig()), LegacyStrategy)
|
||||||
|
|
||||||
|
|
||||||
def test_create_strategy_unknown_raises():
|
def test_create_strategy_unknown_raises():
|
||||||
|
|||||||
Reference in New Issue
Block a user