suaaa7
diff --git a/‎python_ci/src/config.py
Lines changed: 44 additions & 0 deletions b/‎python_ci/src/config.py
Lines changed: 44 additions & 0 deletions
diff --git a/‎python_ci/src/tests/test_config.py
Lines changed: 41 additions & 0 deletions b/‎python_ci/src/tests/test_config.py
Lines changed: 41 additions & 0 deletions
@@ -0,0 +1,44 @@
+from dataclasses import dataclass
+
+
+@dataclass(frozen=True)
+class BaseConfig:
+    version: str
+    model_path: str
+    model_s3_bucket: str
+    model_file_name: str
+
+    def generate_model_path(self) -> str:
+        return "{}/{}/{}".format(
+            self.model_path,
+            self.version,
+            self.model_file_name
+        )
+
+    def generate_model_s3_path(self) -> str:
+        return "s3://{}/{}/{}".format(
+            self.model_s3_bucket,
+            self.version,
+            self.model_file_name
+        )
+
+
+@dataclass(frozen=True)
+class Train:
+    batch_size: int = 16
+    epoch: int = 10
+
+
+@dataclass(frozen=True)
+class Test:
+    batch_size: int = 16
+
+
+@dataclass(frozen=True)
+class Config(BaseConfig):
+    version: str = "v1.0.0"
+    model_path: str = "/ops/models"
+    model_s3_bucket: str = "models"
+    model_file_name: str = "model.pth"
+    train: Train = Train()
+    test: Test = Test()
@@ -0,0 +1,41 @@
+from dataclasses import FrozenInstanceError
+from unittest import TestCase, main
+
+from src.config import BaseConfig, Config
+
+
+class TestConfig(TestCase):
+    def test_generate_path(self):
+        config = BaseConfig(
+            version="v0.0.0",
+            model_path="/opt/models",
+            model_s3_bucket="models",
+            model_file_name="model.pth"
+        )
+
+        expected_model_path = "/opt/models/v0.0.0/model.pth"
+        expected_model_s3_path = "s3://models/v0.0.0/model.pth"
+
+        self.assertEqual(config.generate_model_path(), expected_model_path)
+        self.assertEqual(config.generate_model_s3_path(), expected_model_s3_path)
+
+    def test_config_can_call_method(self):
+        config = Config()
+        config.generate_model_path()
+        config.generate_model_s3_path()
+
+    def test_config_is_immutable(self):
+        config = Config()
+
+        with self.assertRaises(FrozenInstanceError):
+            config.version = "v0.0.0"
+
+        with self.assertRaises(FrozenInstanceError):
+            config.train.epoch = 10000
+
+        with self.assertRaises(FrozenInstanceError):
+            config.test.batch_siz = 64
+
+
+if __name__ == '__main__':
+    main()