示例#1
0
 def test_gym_add_customer(self):
     Gym.id = 1
     g = Gym()
     c = Customer("Pesho", "addr.", "*****@*****.**")
     g.add_customer(c)
     g.add_customer(c)
     self.assertEqual(g.customers, [c])
示例#2
0
 def test_customer_init(self):
     Customer.id = 1
     c = Customer("Pesho", "addr.", "*****@*****.**")
     self.assertEqual(c.id, 1)
     self.assertEqual(c.name, "Pesho")
     self.assertEqual(c.address, "addr.")
     self.assertEqual(c.email, "*****@*****.**")
示例#3
0
 def test_gym_subscription_info(self):
     Gym.id = 1
     Subscription.id = 1
     ExercisePlan.id = 1
     Equipment.id = 1
     Trainer.id = 1
     Customer.id = 1
     g = Gym()
     s = Subscription("10.02.2020", 1, 1, 1)
     p = ExercisePlan(1, 1, 10)
     e = Equipment("Pesho")
     t = Trainer("Pesho")
     c = Customer("Pesho", "addr.", "*****@*****.**")
     g.add_subscription(s)
     g.add_customer(c)
     g.add_equipment(e)
     g.add_plan(p)
     g.add_trainer(t)
     self.assertEqual(
         g.subscription_info(1),
         "Subscription <1> on 10.02.2020\nCustomer <1> Pesho; Address: addr.; Email: [email protected]\nTrainer <1> Pesho\nEquipment <1> Pesho\nPlan <1> with duration 10 minutes"
     )
示例#4
0
from Gym.customer import Customer
from Gym.dvd import DVD
from Gym.movie_world import MovieWorld

c1 = Customer("John", 16, 1)
c2 = Customer("Anna", 55, 2)

d1 = DVD("Black Widow", 1, 2020, "April", 18)
d2 = DVD.from_date(2, "The Croods 2", "23.12.2020", 3)

movie_world = MovieWorld("The Best Movie Shop")

movie_world.add_customer(c1)
movie_world.add_customer(c2)

movie_world.add_dvd(d1)
movie_world.add_dvd(d2)

print(movie_world.rent_dvd(1, 1))
print(movie_world.rent_dvd(2, 1))
print(movie_world.rent_dvd(1, 2))

print(movie_world)
示例#5
0
 def test_customer_repr(self):
     Customer.id = 1
     c = Customer("Pesho", "addr.", "*****@*****.**")
     self.assertEqual(
         str(c),
         "Customer <1> Pesho; Address: addr.; Email: [email protected]")
示例#6
0
 def test_customer_static_method(self):
     Customer.id = 1
     self.assertEqual(Customer.get_next_id(), 1)