demo purpose

DigitalBiomarkerDiscoveryPipeline · Sep 25, 2023 · eea5ef3 · eea5ef3
1 parent 882b277
commit eea5ef3
Show file tree

Hide file tree

Showing 4 changed files with 36 additions and 8 deletions.
diff --git a/dbdpy/__pycache__/device.cpython-39.pyc b/dbdpy/__pycache__/device.cpython-39.pyc
diff --git a/dbdpy/device.py b/dbdpy/device.py
@@ -17,11 +17,26 @@ def get_device_info(self):
         device_info = f"{self.brand} - {self.model}"
         return device_info
 
-    # def calculate_rescale_value(self, age, is_damaged):
-    #     initial_value = 1000
-    #     depre_rate = 0.20 if self.brand == "Apple" else 0.15
-    #     rescale_value = initial_value * ((1 - depre_rate) ** age)
-    #     if is_damaged:
-    #         rescale_value *= 0.70
-
-    #     return round(rescale_value, 2)
+    def calculate_rescale_value(self, age, is_damaged):
+        """Short descreption
+
+        Long description
+
+        Parameters
+        ----------
+        age : int
+            Description for parameter
+        is_damaged : bool
+            Description
+
+        Return
+        ------
+
+        """
+        initial_value = 1000
+        depre_rate = 0.20 if self.brand == "Apple" else 0.15
+        rescale_value = initial_value * ((1 - depre_rate) ** age)
+        if is_damaged:
+            rescale_value *= 0.70
+
+        return round(rescale_value, 2)
diff --git a/tests/__pycache__/test_device.cpython-39-pytest-7.1.2.pyc b/tests/__pycache__/test_device.cpython-39-pytest-7.1.2.pyc
diff --git a/tests/test_device.py b/tests/test_device.py
@@ -18,3 +18,16 @@ def test_get_device_info():
 # def test_calculate_rescale_value(brand, age, is_damaged, expected):
 #     dev = device.CommercialDevice(brand, "AnyModel")
 #     assert dev.calculate_rescale_value(age, is_damaged) == expected
+
+
+@pytest.mark.parametrize(
+    "brand, age, is_damaged, expected",
+    [
+        ("Apple", 1, False, 800),
+        ("Apple", 2, False, 640),
+        ("Fitbit", 1, False, 850),
+    ],
+)
+def test_calculate_rescale_value(brand, age, is_damaged, expected):
+    dev = device.CommercialDevice(brand, "AnyModel")
+    assert dev.calculate_rescale_value(age, is_damaged) == expected