Overflow and Type Errors added to concept.py . Required tests for qua…

.github/pull_request_template.md

@@ -7,9 +7,9 @@
 ## Changes Made
 <!-- List the key changes made in this PR -->
 
-- 
-- 
-- 
+-
+-
+-
 
 ## Testing
 <!-- Describe how these changes were tested -->
@@ -27,4 +27,3 @@
 
 ## Additional Notes
 <!-- Add any other context about the PR here -->
-

.gitignore

@@ -164,3 +164,4 @@ scrap/
 .DS_Store
 .vscode/
 .ruff_cache/
+.python-version

healthchain/models/data/concept.py

@@ -1,5 +1,5 @@
 from enum import Enum
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, Field, field_validator
 from typing import Optional, Dict, Union
 
 
@@ -21,6 +21,25 @@ class Quantity(DataType):
     value: Optional[Union[str, float]] = None
     unit: Optional[str] = None
 
+    @field_validator("value")
+    @classmethod
+    def validate_value(cls, value: Union[str, float]):
+        if value is None:
+            raise TypeError(
+                f"Value CANNOT be a {type(value)} object. Must be float or string in float format."
+            )
+
+        try:
+            return float(value)
+
+        except ValueError:
+            raise ValueError(f"Invalid value '{value}' . Must be a float Number.")
+
+        except OverflowError:
+            raise OverflowError(
+                "Invalid value . Value is too large resulting in overflow."
+            )
+
 
 class Range(DataType):
     low: Optional[Quantity] = None

tests/test_quantity_class.py

@@ -0,0 +1,48 @@
+import pytest
+from healthchain.models.data.concept import Quantity
+from pydantic import ValidationError
+
+
+# Valid Cases
+def test_valid_float_and_integer():
+    valid_floats = [1.0, 0.1, 4.0, 5.99999, 12455.321, 33, 1234]
+    for num in valid_floats:
+        q = Quantity(value=num, unit="mg")
+        assert q.value == num
+
+
+def test_valid_string():
+    valid_strings = ["100", "100.000001", ".1", "1.", ".123", "1234.", "123989"]
+    for string in valid_strings:
+        q = Quantity(value=string, unit="mg")
+        assert q.value == float(string)
+
+
+# Invalid Cases
+def test_invalid_strings():
+    invalid_strings = [
+        "1.0.0",
+        "1..123",
+        "..123",
+        "12..",
+        "12a.56",
+        "1e4.6",
+        "12#.45",
+        "12.12@3",
+        "12@3",
+    ]
+    for string in invalid_strings:
+        with pytest.raises(ValidationError) as exception_info:
+            Quantity(value=string, unit="mg")
+        assert "Invalid value" in str(exception_info.value)
+
+
+# Edge Cases
+def test_edge_cases():
+    edge_cases = ["", "None", None]
+    for val in edge_cases:
+        with pytest.raises((ValidationError, TypeError)) as exception_info:
+            Quantity(value=val, unit="mg")
+
+        exception_info_str = str(exception_info.value)
+        assert any(msg in exception_info_str for msg in ["CANNOT", "Invalid value"])