def test_eval_1(self):
army_list = []
reset_support()
army_list.append(diplomacy_create_army("A", "London", "Support", "B"))
army_list.append(diplomacy_create_army("B", "Paris", "Hold", ""))
army_list = diplomacy_eval(army_list)
self.assertEqual(army_list[0].location, "London")
self.assertEqual(army_list[1].location, "Paris")
def test_eval_2(self):
army_list = []
reset_support()
army_list.append(diplomacy_create_army("A", "Madrid", "Hold", ""))
army_list.append(
diplomacy_create_army("B", "Barcelona", "Move", "Madrid"))
army_list = diplomacy_eval(army_list)
self.assertEqual(army_list[0].location, "[dead]")
self.assertEqual(army_list[1].location, "[dead]")
def test_eval_3(self):
v = diplomacy_eval([["A", "Madrid", "Hold"],
["B", "Barcelona", "Move", "Madrid"],
["C", "London", "Move", "Madrid"],
["D", "Paris", "Support", "B"]])
self.assertEqual(
v, {
"Madrid": [["A", "[dead]"], ["B", 1], ["C", "[dead]"]],
"Paris": [["D", 0]]
})