Change LARGE to VehicleSize.LARGE in parking lot solution (donnemarti…

…n#146)
Godfrey-Ndungu · Apr 9, 2018 · f552598 · f552598
1 parent 7e316ce
commit f552598
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/solutions/object_oriented_design/parking_lot/parking_lot.py b/solutions/object_oriented_design/parking_lot/parking_lot.py
@@ -45,7 +45,7 @@ def __init__(self, license_plate):
         super(Car, self).__init__(VehicleSize.COMPACT, license_plate, spot_size=1)
 
     def can_fit_in_spot(self, spot):
-        return True if (spot.size == LARGE or spot.size == COMPACT) else False
+        return spot.size in (VehicleSize.LARGE, VehicleSize.COMPACT)
 
 
 class Bus(Vehicle):
@@ -54,7 +54,7 @@ def __init__(self, license_plate):
         super(Bus, self).__init__(VehicleSize.LARGE, license_plate, spot_size=5)
 
     def can_fit_in_spot(self, spot):
-        return True if spot.size == LARGE else False
+        return spot.size == VehicleSize.LARGE
 
 
 class ParkingLot(object):