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)
