def testToString(self):
oilField = OilField()
oilField.Name = "Frigg"
t0 = datetime.now() - timedelta(days=2)
t1 = datetime.now() - timedelta(days=1)
t2 = datetime.now()
oilField.ProductionSpans.append(ProductionSpan.instance(t0, t1, 10))
oilField.ProductionSpans.append(ProductionSpan.instance(t1, t2, 6))
print oilField
def testTotalProduction(self):
oilField = OilField()
t0 = datetime.now() - timedelta(days=2)
t1 = datetime.now() - timedelta(days=1)
t2 = datetime.now()
oilField.ProductionSpans.append(ProductionSpan.instance(t0, t1, 10))
oilField.ProductionSpans.append(ProductionSpan.instance(t1, t2, 6))
self.assertEqual(oilField.totalProduction(), 16)
self.assertEqual(oilField.discovery(), t0)
