def test_size_constant(self):
constraint = SizeRandomSoftConstraint(10, 10, 10)
seminar = self.create_seminar(1, 2)
hugs = seminar.camps[0].hugs
educative1 = Educative()
educative1.hug = hugs[0]
educative2 = Educative()
educative2.hug = hugs[0]
educative3 = Educative()
educative3.hug = hugs[1]
score = constraint.calculate_score([educative1, educative2, educative3], seminar)
self.assertEquals(score, 50)
def test_gender_constant(self):
constraint = GenderRandomSoftConstraint(10, 10, 10)
seminar = self.create_seminar(1, 2)
hugs = seminar.camps[0].hugs
educative1 = Educative()
educative1.gender = MALE
educative1.hug = hugs[0]
educative2 = Educative()
educative2.gender = FEMALE
educative2.hug = hugs[0]
educative3 = Educative()
educative3.gender = FEMALE
educative3.hug = hugs[1]
score = constraint.calculate_score([educative1, educative2, educative3], seminar)
self.assertEquals(score, 30)
def test_vegetarian(self):
aviad = Educative()
aviad.first_name = 'Aviad'
aviad.food = VEGETARIAN
self.session.add(aviad)
seminar = self.create_seminar(1, 1)
hug = seminar.camps[0].hugs[0]
hug.food = MEAT
aviad.hug = hug
constraint = VegetarianHardConstraint()
self.assertTrue(not constraint.is_valid(aviad, [aviad], [hug]))
def test_ken_constant(self):
constraint = KenRandomSoftConstraint(10, 10, 10)
seminar = self.create_seminar(1, 2)
hugs = seminar.camps[0].hugs
ken1 = Ken()
ken1.name = "A"
ken2 = Ken()
ken2.name = "B"
second_ken = SecondKen()
second_ken.name = "c"
self.session.add(ken1)
self.session.add(ken2)
self.session.add(second_ken)
educative1 = Educative()
educative1.ken = ken1
educative1.hug = hugs[0]
self.session.add(educative1)
educative2 = Educative()
educative2.ken = ken1
educative2.hug = hugs[0]
self.session.add(educative2)
educative3 = Educative()
educative3.ken = ken1
educative3.hug = hugs[1]
self.session.add(educative3)
educative4 = Educative()
educative4.ken = ken2
educative4.second_ken = second_ken
educative4.hug = hugs[1]
self.session.add(educative4)
score = constraint.calculate_score([educative1, educative2, educative3, educative4], seminar)
self.assertEquals(score, 70)
def test_madrich(self):
ken = Ken()
ken.name = 'Misgav'
seminar = self.create_seminar(1, 1)
hug = seminar.camps[0].hugs[0]
aviad = Educative()
aviad.first_name = 'Aviad'
aviad.ken = ken
aviad.hug = hug
madrich = Madrich()
madrich.first_name = 'Cool'
madrich.last_name = 'Guy'
madrich.ken = ken
madrich.hug = hug
constraint = MadrichHardConstraint()
self.assertTrue(not constraint.is_valid(aviad, [aviad], seminar))
