def test_amount_needed_calculation_correct(self): new_frame = v.add_amount_needed(df, 'goal', 'raised') amount_n = new_frame['amount_needed'] self.assertEquals(amount_n[0], 2) self.assertEquals(amount_n[5], 0)
def test_if_col_not_exist_keyerror(self): self.assertRaises(KeyError, v.add_amount_needed(df, 'qwerty1', 'qwerty2'))