mirror of
https://github.com/vale981/ray
synced 2025-03-09 04:46:38 -04:00
31 lines
878 B
Python
31 lines
878 B
Python
|
"""Testing for trainer class"""
|
||
|
|
import copy
|
||
|
|
import unittest
|
||
|
|
from ray.rllib.agents.trainer import Trainer, COMMON_CONFIG
|
||
|
|
|
||
|
|
|
||
|
|
class TestTrainer(unittest.TestCase):
|
||
|
|
def test_validate_config_idempotent(self):
|
||
|
|
"""
|
||
|
|
Asserts that validate_config run multiple
|
||
|
|
times on COMMON_CONFIG will be idempotent
|
||
|
|
"""
|
||
|
|
# Given
|
||
|
|
standard_config = copy.deepcopy(COMMON_CONFIG)
|
||
|
|
standard_config["_use_trajectory_view_api"] = False
|
||
|
|
|
||
|
|
# When (we validate config 2 times)
|
||
|
|
Trainer._validate_config(standard_config)
|
||
|
|
config_v1 = copy.deepcopy(standard_config)
|
||
|
|
Trainer._validate_config(standard_config)
|
||
|
|
config_v2 = copy.deepcopy(standard_config)
|
||
|
|
|
||
|
|
# Then
|
||
|
|
self.assertEqual(config_v1, config_v2)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
import pytest
|
||
|
|
import sys
|
||
|
|
sys.exit(pytest.main(["-v", __file__]))
|