def test_complicated(self):
self.frein = Frein.objects.create(lettre_commande=self.lc,
pro1=Frein.PRO_EXPE,
pro2 = Frein.PRO_PP)
expected_result = LCFreinResult()
expected_result.add_first(Frein.PRO_INTIT, Frein.PRO_EXPE)
expected_result.add(Frein.PRO_INTIT, Frein.PRO_PP)
self.assertEqual(expected_result , LCFreinCalc(self.lc).calculate())
def test_add(self):
frein_categorie = Frein.PRO_INTIT
frein_valeur = Frein.PRO_ETU
expected_result = LCFreinResult()
expected_result.frein_result_dic[frein_categorie].all_frein[frein_valeur] += 1
produced_result = LCFreinResult()
produced_result.add(frein_categorie, frein_valeur)
self.assertEqual(expected_result, produced_result)
def test_addition(self):
lc_frein_result_1 = LCFreinResult()
lc_frein_result_1.add(Frein.PRO_INTIT, Frein.PRO_ETU)
lc_frein_result_2 = LCFreinResult()
lc_frein_result_2.add(Frein.PRO_INTIT, Frein.PRO_ETU)
expected_result = LCFreinResult()
expected_result.add(Frein.PRO_INTIT, Frein.PRO_ETU)
expected_result.add(Frein.PRO_INTIT, Frein.PRO_ETU)
self.assertEqual(expected_result, lc_frein_result_1 + lc_frein_result_2)
