class GetQueryParametersTestCase(EngineerTestCase):
"""
Tests methods for retrieving query parameters.
"""
def setUp(self):
super(GetQueryParametersTestCase, self).setUp()
self.engineer = Engineer(query=self.query, specs=PipeSpecSheet())
def test_get_accounts(self):
"""
Tests the get_accounts method.
"""
accounts = self.engineer.get_accounts()
self.assertEqual(len(accounts), 2)
def test_get_locations(self):
"""
Tests the get_locations method.
"""
locations = self.engineer.get_locations()
self.assertEqual(len(locations), 4)
def test_get_searchterms(self):
"""
Tests the get_searchterms method.
"""
searchterms = self.engineer.get_searchterms()
self.assertEqual(len(searchterms), 3)
def test_format_for_box(self):
"""
Tests the format_locations method for a 'box' location_format.
"""
query = ReservoirQuery(locations=self.locations)
specs = PipeSpecSheet(location_format='box')
engineer = Engineer(query=query, specs=specs)
engineer._format_locations()
all_boxes = False
for new_location in engineer.get_locations():
if new_location.shape == 'Rectangle':
all_boxes = True
else:
all_boxes = False
break
self.assertTrue(all_boxes)
def test_format_for_radius(self):
"""
Tests the format_locations method for a 'radius' location_format.
"""
query = ReservoirQuery(locations=self.locations)
specs = PipeSpecSheet(location_format='radius', radius_limit_km=100)
engineer = Engineer(query=query, specs=specs)
engineer._format_locations()
all_circles = False
for new_location in engineer.get_locations():
if new_location.shape == 'Circle':
all_circles = True
else:
all_circles = False
break
self.assertTrue(all_circles)
