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_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)