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")
def test_gym_add_subscription(self):
g = Gym()
s = Subscription("10.02.2020", 1, 1, 1)
g.add_subscription(s)
g.add_subscription(s)
self.assertEqual(g.subscriptions, [s])
from project.subscription import Subscription
from project.trainer import Trainer
customer = Customer("John", "Maple Street", "*****@*****.**")
equipment = Equipment("Treadmill")
trainer = Trainer("Peter")
subscription = Subscription("14.05.2020", 1, 1, 1)
plan = ExercisePlan(1, 1, 20)
gym = Gym()
gym.add_customer(customer)
gym.add_equipment(equipment)
gym.add_trainer(trainer)
gym.add_plan(plan)
gym.add_subscription(subscription)
print(Customer.get_next_id())
print(gym.subscription_info(1))
# c = Customer("Sofia", "Lozen", "*****@*****.**")
# print(c)
#
# m = Customer("Sofia", "Lozen", "*****@*****.**")
# print(c)
# print(m)
# equip1 = Equipment("Number 1")
# equip2 = Equipment("Number 2")
# print(equip1)
# print(equip2)
