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_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'))
def test_if_col_not_exist_keyerror(self):
self.assertRaises(KeyError, v.add_amount_needed(df, 'qwerty1', 'qwerty2'))
