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)