def test_next_month(self): """ Tests that month increment is working. """ bseeRequest = UsRequest() self.assertEqual(bseeRequest.year_month, date(year=1947, month=1, day=1)) bseeRequest.nextMonth() self.assertEqual(bseeRequest.year_month, date(year=1947, month=2, day=1))
def test_get_soup(self): """ Tests that production is correct. """ usRequest = UsRequest() usRequest.year_month = date(year=2013, month=1, day=1) soup = usRequest.get_soup(page=1) table = soup.find('table', border=5, width=600) trs = table.findAll('tr') self.assertEqual(len(trs), 1002)
def test_get_production(self): """ Tests that production is correct. """ soup = BeautifulSoup("<tr><td>G03229</td><td>5</td><td>2013</td><td>304</td>" + "<td>0</td><td>0</td><td>516</td><td>13,129</td><td>1</td><td>18</td></tr>") bseeRequest = UsRequest() production = bseeRequest.get_production(soup) expectedProduction = FieldProduction( name="G03229", country="US", date=date(year=2013, month=5, day=1), production_oil=304, production_gas=516, depth=18, ) self.assertEqual(production.name, expectedProduction.name) self.assertEqual(production.country, expectedProduction.country) self.assertEqual(production.date, expectedProduction.date) self.assertEqual(production.production_oil, expectedProduction.production_oil) self.assertEqual(production.production_gas, expectedProduction.production_gas) self.assertEqual(production.depth, expectedProduction.depth)
def test_get_productions(self): """ Tests that production is correct. """ usRequest = UsRequest() usRequest.year_month = date(year=2013, month=1, day=1) productions = usRequest.getProductions() self.assertEqual(len(productions), 1302) production = productions[0] expectedProduction = FieldProduction( name="G03205", country="US", date=date(year=2013, month=1, day=1), production_oil=17271, production_gas=9446, depth=495, ) self.assertEqual(production.name, expectedProduction.name) self.assertEqual(production.country, expectedProduction.country) self.assertEqual(production.date, expectedProduction.date) self.assertEqual(production.production_oil, expectedProduction.production_oil) self.assertEqual(production.production_gas, expectedProduction.production_gas) self.assertEqual(production.depth, expectedProduction.depth) production = productions[-1] expectedProduction = FieldProduction( name="G15212", country="US", date=date(year=2013, month=1, day=1), production_oil=4773+34, production_gas=3626+9121, depth=140, ) self.assertEqual(production.name, expectedProduction.name) self.assertEqual(production.country, expectedProduction.country) self.assertEqual(production.date, expectedProduction.date) self.assertEqual(production.production_oil, expectedProduction.production_oil) self.assertEqual(production.production_gas, expectedProduction.production_gas) self.assertEqual(production.depth, expectedProduction.depth)