def setUp(self): self.well = Well(data={"WOPT": [1, 1, 2, 4, 5]}) self.well.add_parameter({"WWPT": [1, 1, 2, 4, 5]}) self.well.add_parameter({"WLPR": [1, 1, 1, 2, 4]})
class Test(unittest.TestCase): def setUp(self): self.well = Well(data={"WOPT": [1, 1, 2, 4, 5]}) self.well.add_parameter({"WWPT": [1, 1, 2, 4, 5]}) self.well.add_parameter({"WLPR": [1, 1, 1, 2, 4]}) def tearDown(self): del (self.well) # def test_add_well(self): # self.assertTrue(self.well) # self.assertEqual(self.well.classification, [2, 4, 2]) def test_add_worktime(self): self.well.add_worktime() self.assertEqual(self.well.work_time, [12, 12]) @unittest.skip("demonstrating skipping") def test_completion_year(self): self.well.add_worktime() self.well.completion_year() self.assertEqual(self.well.first_run, 0) @unittest.skip("demonstrating skipping") def test_abandonment_year(self): self.well.add_worktime() self.well.abandonment_year() self.assertEqual(self.well.abandonment, "working") @unittest.skip("demonstrating skipping") def test_classification(self): self.well.add_worktime() self.well.completion_year() self.well.abandonment_year() self.well.well_classification() self.assertEqual(self.well.classification, [2, 4, 2, 2]) @unittest.skip("demonstrating skipping") def test_classification_rate(self): self.well.add_worktime() self.well.completion_year() self.well.abandonment_year() self.well.well_classification(mode="rate") self.assertEqual(self.well.classification_by_rate, [2, 4, 2, 2]) def test_compress_data(self): self.well.add_worktime({"1990": 0, "1991": 2, "1992": 4}) self.well.compress_data({"1990": 0, "1991": 2, "1992": 4}) self.assertEqual(self.well.parameters["WOPT"], [1, 3]) self.assertEqual(self.well.parameters["WWPT"], [1, 3]) self.assertEqual(self.well.parameters["WLPR"], [1, 4])