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)
