super-gradients/tests/unit_tests/training_utils_test.py

import unittest

from training.utils.utils import recursive_override


class TestTrainingUtils(unittest.TestCase):
    def test_recursive_override(self):
        base_dict = {"a": 1, "b": 2, "c": {
            "x": 10, "y": 20, "z": {
                "q": "q_str", "i": "i_str"
            }
        }}

        ext_dict = {"b": 4, "c": {
            "x": 20, "z": {
                "q": "q_str_new"
            }
        }}

        recursive_override(base_dict, ext_dict)

        self.assertEqual(base_dict["a"], 1)
        self.assertEqual(base_dict["b"], 4)
        self.assertEqual(base_dict["c"]["x"], 20)
        self.assertEqual(base_dict["c"]["y"], 20)
        self.assertEqual(base_dict["c"]["z"]["q"], "q_str_new")
        self.assertEqual(base_dict["c"]["z"]["i"], "i_str")


if __name__ == '__main__':
    unittest.main()