test added

checklist.md

@@ -2,7 +2,8 @@
 
 ## Code Style
 
-1. If you have some long math, you can split it onto additional variables.
+1. If you have some long math, you can split it onto additional variables, 
+or break after binary operations (not before - it cause the W504 errors)
 
 Good example:
 
@@ -13,6 +14,18 @@ estimated_time = distance_to_the_destinatoin / estimated_speed
 how_much_fuel_needed = fuel_consumption * estimated_time * overlap_coeficient
 ```
 
+Good example:
+
+```python
+how_much_fuel_needed = (max_fuel_consumption
+                        * height_fuel_consumption_coeficient
+                        * distance_to_the_destinatoin
+                        / (plan_max_speed
+                           - wind_awerage_speed
+                           * wind_angle_coefisient)
+                        * overlap_coeficient)
+```
+
 Bad example:
 
 ```python
@@ -25,20 +38,18 @@ how_much_fuel_needed = max_fuel_consumption \
                        ) * overlap_coeficient
 ```
 
-## Code Efficiency
-
 2. Use descriptive and correct variable names.
 
 Good example:
 
 ```python
-def get_full_name(first_name: str, last_name: str):
+def get_full_name(first_name: str, last_name: str) -> str:
     return f"{first_name} {last_name}"
 ```
 
 Bad example:
 ```python
-def get_full_name(x: str, y: str):
+def get_full_name(x: str, y: str) -> str:
     return f"{x} {y}"
 ```
 
@@ -49,7 +60,7 @@ def get_full_name(x: str, y: str):
 Good example:
 
 ```python
-def is_adult(age):
+def is_adult(age: int) -> str:
     if age >= 18:
         return "adult"
     return "not an adult"
@@ -58,7 +69,7 @@ def is_adult(age):
 Bad example:
 
 ```python
-def is_adult(age):
+def is_adult(age: int) -> str:
     if age >= 18:
         return "adult"
     else:

tests/test_main.py

@@ -1,3 +1,5 @@
+from unittest.mock import patch
+
 import pytest
 
 from app.main import Car, CarWashStation
@@ -37,6 +39,12 @@ def test_car_wash_station(cars, wash_station, total_cost):
     income = wash_station.serve_cars(cars)
     assert income == total_cost, f"Income should equal to {total_cost}"
 
+def test_wash_single_car_is_called():
+    with patch.object(CarWashStation, 'wash_single_car') as mock_method:
+        CarWashStation(3, 9, 4, 11).serve_cars([Car(2, 1, "Ford")])
+        assert mock_method.called, "Expected 'wash_single_car' to have " \
+                                   "been called inside 'serve_cars' method"
+
 
 @pytest.mark.parametrize(
     "cars,wash_station,cars_clean_marks",