Browse Source

Change LARGE to VehicleSize.LARGE in parking lot solution (#146)

cclauss 7 years ago
parent
commit
1b339dbdde
1 changed files with 2 additions and 2 deletions
  1. 2 2
      solutions/object_oriented_design/parking_lot/parking_lot.py

+ 2 - 2
solutions/object_oriented_design/parking_lot/parking_lot.py

@@ -45,7 +45,7 @@ class Car(Vehicle):
         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 @@ class Bus(Vehicle):
         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):