class BalanceTheBooksStrategyTest(unittest.TestCase): """ Test suite for balance_the_books_strategy.py.""" def setUp(self): self.decision = Decision() self.member = Member() self.bill = Bill() self.strategy = BalanceTheBooksStrategy(self.decision, self.member, self.bill) self.stance = Stance() self.stance.issue = "Some Issue" self.stance.side = outcomes.FOR self.stance1 = Stance() self.stance1.issue = "Some Other Issue" self.stance1.side = outcomes.AGN self.stance2 = Stance() self.stance2.issue = "Another Issue" self.stance2.side = outcomes.FOR self.decision.for_stances = [self.stance] self.decision.agn_stances = [self.stance1] def test_init(self): """Tests whether a strategy can be constructed properly""" result = BalanceTheBooksStrategy(self.decision, self.member, self.bill) self.assertEqual(result._name, "Balance the Books") self.assertEqual(result._decision, self.decision) self.assertEqual(result._member, self.member) self.assertEqual(result._bill, self.bill) self.assertEqual(result._success, False) def test_run_fail_no_majority(self): """ Verifies the function fails if there is no majority.""" self.decision.for_stances = [self.stance] self.decision.agn_stances = [self.stance1] self.decision.split_record = [self.stance, self.stance1] result = self.strategy.run() self.assertFalse(result) self.assertFalse(self.strategy._success) self.assertEquals(self.decision.result, None) self.assertEquals(self.decision.reason, None) def test_run_fail_no_split_record(self): """ Verifies the function fails if the member's record is not split on the decision.""" self.decision.for_stances = [self.stance, self.stance2] self.decision.agn_stances = [self.stance1] result = self.strategy.run() self.assertFalse(result) self.assertFalse(self.strategy._success) self.assertEquals(self.decision.result, None) self.assertEquals(self.decision.reason, None) def test_run_success_FOR(self): """ Verifies that run() successfully makes a FOR decision""" self.decision.for_stances = [self.stance, self.stance2] self.decision.agn_stances = [self.stance1] self.decision.split_record = [self.stance, self.stance1] result = self.strategy.run() self.assertTrue(result) self.assertTrue(self.strategy._success) self.assertEquals(self.decision.result, outcomes.FOR) self.assertEquals(self.decision.reason, self.decision.for_stances) self.assertEquals(self.decision.downside, self.decision.agn_stances) def test_run_success_AGN(self): """ Verifies that run() successfully makes a AGN decision""" self.decision.for_stances = [self.stance] self.decision.agn_stances = [self.stance1, self.stance2] self.decision.split_record = [self.stance, self.stance1] result = self.strategy.run() self.assertTrue(result) self.assertTrue(self.strategy._success) self.assertEquals(self.decision.result, outcomes.AGN) self.assertEquals(self.decision.reason, self.decision.agn_stances) self.assertEquals(self.decision.downside, self.decision.for_stances) def test_explain(self): """ Verifies explain runs if there is a success [aka _explain is implemented].""" self.decision.for_stances = [self.stance] self.decision.agn_stances = [self.stance1, self.stance2] self.decision.split_record = [self.stance, self.stance1] result = self.strategy.run() self.assertTrue(result) self.strategy.explain()