From f552598088c7ec25d63df40e8a218e507f4bdcf3 Mon Sep 17 00:00:00 2001 From: cclauss Date: Mon, 9 Apr 2018 02:25:11 +0200 Subject: [PATCH] Change LARGE to VehicleSize.LARGE in parking lot solution (#146) --- solutions/object_oriented_design/parking_lot/parking_lot.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/solutions/object_oriented_design/parking_lot/parking_lot.py b/solutions/object_oriented_design/parking_lot/parking_lot.py index 9167d9fdb7..12fcd24596 100644 --- 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):