tests/test_config.py

# import json
# import unittest

# from src.data.config import NORMALIZATION_DICT_FILENAME
# from src.data.dataset import Normalizer
# from src.galileo import Encoder
# from src.utils import check_config, config_dir, load_check_config


# class TestConfigs(unittest.TestCase):
#     @staticmethod
#     def check_models_can_be_loaded(config):
#         _ = Encoder(**config["model"]["encoder"])

#     def test_configs(self):
#         configs = list((config_dir / "mae").glob("*.json"))

#         for config_path in configs:
#             try:
#                 loaded_config = load_check_config(config_path.name)
#                 self.check_models_can_be_loaded(loaded_config)
#             except Exception as e:
#                 print(f"Failed for {config_path} with {e}")
#                 raise e

#     def test_normalization_dict(self):
#         if (config_dir / NORMALIZATION_DICT_FILENAME).exists():
#             with (config_dir / NORMALIZATION_DICT_FILENAME).open("r") as f:
#                 norm_dict = json.load(f)
#         output_dict = {}
#         for key, val in norm_dict.items():
#             if "n" not in key:
#                 output_dict[int(key)] = val
#             else:
#                 output_dict[key] = val
#         normalizer = Normalizer(std=True, normalizing_dicts=output_dict)
#         for key, val in normalizer.shift_div_dict.items():
#             divs = val["div"]
#             for d in divs:
#                 self.assertNotEqual(d, 0, f"0 in {key}")