Skip to content

Commit

Permalink
fix test & flake errors
Browse files Browse the repository at this point in the history
  • Loading branch information
artem.puhach committed Nov 20, 2023
1 parent bdc721b commit 4e6c7dc
Show file tree
Hide file tree
Showing 5 changed files with 16 additions and 13 deletions.
2 changes: 1 addition & 1 deletion app/cinema/bar.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,4 +7,4 @@ class CinemaBar:

@staticmethod
def sell_product(product: str, customer: Customer) -> None:
print(f"Cinema bar sold {product} to {customer}.")
print(f"Cinema bar sold {product} to {customer.name}.")
8 changes: 4 additions & 4 deletions app/cinema/hall.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,17 +3,17 @@


class CinemaHall:
def __init__(self, number):
def __init__(self, number: int) -> None:
self.number = number

def movie_session(
self,
movie_name: str,
customers: list[Customer],
cleaning_staff: Cleaner
):
print(f"\"{movie_name}\" started in hall number {self.number}.")
) -> None:
print(f'\"{movie_name}\" started in hall number {self.number}.')
for customer in customers:
customer.watch_movie(movie_name)
print(f"\"{movie_name}\" ended.")
print(f'\"{movie_name}\" ended.')
cleaning_staff.clean_hall(self.number)
7 changes: 5 additions & 2 deletions app/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,8 +8,11 @@ def cinema_visit(
customers: list,
hall_number: int,
cleaner: str,
movie: str):
customer_instances = [Customer(customer["name"], customer["food"]) for customer in customers]
movie: str) -> None:
customer_instances = [
Customer(customer["name"], customer["food"])
for customer in customers
]
new_cinema_hall = CinemaHall(hall_number)
new_cinema_bar = CinemaBar()
new_cleaner = Cleaner(cleaner)
Expand Down
6 changes: 3 additions & 3 deletions app/people/cinema_staff.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
class Cleaner:
def __init__(self, name):
def __init__(self, name: str) -> None:
self.name = name

def clean_hall(self, hall_number):
print(f"# Cleaner {self.name} is cleaning hall number {hall_number}.")
def clean_hall(self, hall_number: int) -> None:
print(f"Cleaner {self.name} is cleaning hall number {hall_number}.")
6 changes: 3 additions & 3 deletions app/people/customer.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
class Customer:
def __init__(self, name: str, food: str):
def __init__(self, name: str, food: str) -> None:
self.name = name
self.food = food

def watch_movie(self, movie):
print(f"{self.name} is watching \"{movie}\".")
def watch_movie(self, movie: str) -> None:
print(f'{self.name} is watching \"{movie}\".')

0 comments on commit 4e6c7dc

Please sign in to comment.