def test_hourly(self): curve = PlantSharesCurve(5000) self.assertEqual(25*[5000], list(curve.hourly(isodate('2016-05-01'), isodate('2016-05-01'))))
def test_atDay(self): curve = PlantSharesCurve(5000) self.assertEqual(5000, curve.atDay(isodate('2016-05-01')))