From 206fe8e612f62e84f77f89739d60cbc35471ce3a Mon Sep 17 00:00:00 2001
From: Florian Bezannier <florian.bezannier@hotmail.fr>
Date: Mon, 15 Mar 2021 19:53:18 +0100
Subject: [PATCH] add fct to get car type

---
 Car.py | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/Car.py b/Car.py
index 20172f9..3dabbb5 100644
--- a/Car.py
+++ b/Car.py
@@ -35,6 +35,15 @@ class Car:
             self.battery_power = DEFAULT_BATTERY_POWER
             self.fuel_capacity = DEFAULT_FUEL_CAPACITY
 
+    def is_electric(self) -> bool:
+        return self.fuel_capacity == 0 and self.battery_power > 0
+
+    def is_thermal(self) -> bool:
+        return self.fuel_capacity > 0 and self.battery_power == 0
+
+    def is_hybrid(self) -> bool:
+        return self.fuel_capacity > 0 and self.battery_power > 0
+
     @classmethod
     def from_json(cls, data: dict):
         return cls(**data)