fix : minor edit in tests

sepandhaghighi · Sep 4, 2024 · 5330164 · 5330164
1 parent 9c8c194
commit 5330164
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mycoffee/functions.py b/mycoffee/functions.py
@@ -98,8 +98,8 @@ def coffee_calc(params, digit=3):
     :return: coffee amount as float
     """
     coffee = params["water"] * params["coffee_ratio"] / params["water_ratio"]
-    if is_int(coffee):
-        return coffee
+    if is_int(coffee) or digit == 0:
+        return int(coffee)
     return round(coffee, digit)