def test_remove(self): army = Army() province = Province(None, 1, 1) province.add_army(army) army.remove() self.assertEqual(army.province, None) self.assertFalse(army in province.armies)