#1711 - Added the ability to load legacy lay down config files. Added extensive unit testing and end-to-end testing. Also added the ability to set exactly how many num_train_steps, num_eval_steps, num_train_episodes, and num_eval_episode and used when converting a legacy training config.

tests/test_lay_down_config.py

@@ -0,0 +1,44 @@
+# © Crown-owned copyright 2023, Defence Science and Technology Laboratory UK
+import pytest
+import yaml
+
+from primaite.config.lay_down_config import (
+    convert_legacy_lay_down_config,
+    data_manipulation_config_path,
+    ddos_basic_one_config_path,
+    ddos_basic_two_config_path,
+    dos_very_basic_config_path,
+)
+from tests import TEST_CONFIG_ROOT
+
+
+@pytest.mark.parametrize(
+    "legacy_file, new_path",
+    [
+        ("legacy_config_1_DDOS_BASIC.yaml", ddos_basic_one_config_path()),
+        ("legacy_config_2_DDOS_BASIC.yaml", ddos_basic_two_config_path()),
+        ("legacy_config_3_DOS_VERY_BASIC.yaml", dos_very_basic_config_path()),
+        ("legacy_config_5_DATA_MANIPULATION.yaml", data_manipulation_config_path()),
+    ],
+)
+def test_legacy_lay_down_config_load(legacy_file, new_path):
+    """Tests converting legacy lay down files into the new format."""
+    with open(TEST_CONFIG_ROOT / "legacy_conversion" / legacy_file, "r") as file:
+        legacy_lay_down_config = yaml.safe_load(file)
+
+    with open(new_path, "r") as file:
+        new_lay_down_config = yaml.safe_load(file)
+
+    converted_lay_down_config = convert_legacy_lay_down_config(legacy_lay_down_config)
+
+    assert len(converted_lay_down_config) == len(new_lay_down_config)
+
+    for i, new_item in enumerate(new_lay_down_config):
+        converted_item = converted_lay_down_config[i]
+
+        for key, val in new_item.items():
+            if key == "position":
+                continue
+            assert key in converted_item
+
+            assert val == converted_item[key]