python-mytnb 0.5.0__tar.gz → 0.5.1__tar.gz

{python_mytnb-0.5.0 → python_mytnb-0.5.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: python-mytnb
-Version: 0.5.0
+Version: 0.5.1
 Summary: Python library to interface with the myTNB API (Tenaga Nasional Berhad)
 Project-URL: Repository, https://github.com/danieyal/python-mytnb
 License-Expression: MIT

{python_mytnb-0.5.0 → python_mytnb-0.5.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "python-mytnb"
-version = "0.5.0"
+version = "0.5.1"
 description = "Python library to interface with the myTNB API (Tenaga Nasional Berhad)"
 readme = "README.md"
 license = "MIT"

{python_mytnb-0.5.0 → python_mytnb-0.5.1}/src/mytnb/models.py

@@ -267,6 +267,13 @@ class AccountUsage(BaseModel):
                 return m.numeric_value
         return None
 
+    @property
+    def average_usage_kwh(self) -> Optional[float]:
+        for m in self.usage_metrics:
+            if m.key == "AVGUSAGE":
+                return m.numeric_value
+        return None
+
     @property
     def current_cost_rm(self) -> Optional[float]:
         for m in self.cost_metrics:

{python_mytnb-0.5.0 → python_mytnb-0.5.1}/tests/test_models.py

@@ -244,6 +244,10 @@ class TestAccountUsage:
         usage = AccountUsage.from_api_response(FULL_API_RESPONSE)
         assert usage.current_usage_kwh == 234.5
 
+    def test_average_usage_kwh(self):
+        usage = AccountUsage.from_api_response(FULL_API_RESPONSE)
+        assert usage.average_usage_kwh == 15.6
+
     def test_current_cost_rm(self):
         usage = AccountUsage.from_api_response(FULL_API_RESPONSE)
         assert usage.current_cost_rm == 87.60
@@ -280,6 +284,7 @@ class TestAccountUsage:
     def test_empty_response(self):
         usage = AccountUsage.from_api_response({})
         assert usage.current_usage_kwh is None
+        assert usage.average_usage_kwh is None
         assert usage.current_cost_rm is None
         assert usage.projected_cost_rm is None
         assert usage.by_month is None
@@ -291,6 +296,7 @@ class TestAccountUsage:
         }
         usage = AccountUsage.from_api_response(data)
         assert usage.current_usage_kwh is None
+        assert usage.average_usage_kwh is None
         assert usage.current_cost_rm is None