Преглед изворни кода

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

cclauss пре 7 година
родитељ
комит
1b339dbdde
1 измењених фајлова са 2 додато и 2 уклоњено
  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):