add decorator

Maclovi · May 3, 2024 · 0ba9b60 · 0ba9b60
1 parent a7cc46b
commit 0ba9b60
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/tests/test_dotenv.py b/tests/test_dotenv.py
@@ -1,6 +1,7 @@
 import unittest
 from decimal import Decimal
 
+from src.mapenv.improve import Improve as improve
 from src.mapenv.main import MapEnv
 
 
@@ -9,7 +10,8 @@ def __init__(self, val: str) -> None:
         self.val = val
 
 
-class Telegram(MapEnv, load_env="tests/.env", override=True):
+@improve(envfile="tests/.env", override=True, frozen=True)
+class Telegram(MapEnv):
     TOKEN: str
     FLOAT: float
     SET: set[int]