def test_contributorchart_get_payment_frequencies(self): cc = ContributorChart() groups = cc.get_payment_frequencies() self.assertEqual(groups[0]['payment_freq'], 'every month') self.assertEqual(groups[0]['average_payment_freq'], 29.5) # (28 + 31) / 2 self.assertAlmostEqual(groups[0]['total_paid'], 3*42.4) self.assertEqual(groups[1]['payment_freq'], 'every 3 months') self.assertEqual(groups[1]['average_payment_freq'], 92) # (32 + 29 + 32) / 3 self.assertAlmostEqual(groups[1]['total_paid'], 4*42.4) self.assertEqual(groups[2]['payment_freq'], 'every 3 months') self.assertEqual(groups[2]['average_payment_freq'], None) self.assertAlmostEqual(groups[2]['total_paid'], 42.4) # embed()
def test_contributorchart_get_payments(self): cc = ContributorChart() payments_df = cc.get_payments() self.assertEqual(len(payments_df['date']), 8) self.assertEqual(float(payments_df['amount'].sum()), 8*42.4)