mirrors/system-design-primer
1
0
Fork 0
mirror of https://github.com/donnemartin/system-design-primer.git synced 2025-09-16 17:10:39 +03:00
Code Issues Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
cclauss
2018-04-09 02:25:11 +02:00
committed by Donne Martin
parent ea262de564
commit 1b339dbdde
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
solutions/object_oriented_design/parking_lot/parking_lot.py
View File

@@ -45,7 +45,7 @@ class Car(Vehicle):
super(Car, self).__init__(VehicleSize.COMPACT, license_plate, spot_size=1) super(Car, self).__init__(VehicleSize.COMPACT, license_plate, spot_size=1)
def can_fit_in_spot(self, spot): 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): class Bus(Vehicle):
@@ -54,7 +54,7 @@ class Bus(Vehicle):
super(Bus, self).__init__(VehicleSize.LARGE, license_plate, spot_size=5) super(Bus, self).__init__(VehicleSize.LARGE, license_plate, spot_size=5)
def can_fit_in_spot(self, spot): def can_fit_in_spot(self, spot):
return True if spot.size == LARGE else False return spot.size == VehicleSize.LARGE
class ParkingLot(object): class ParkingLot(object):