def test_find_nearby(self): stations = self.lib.find_nearby(Position(0, 0)) other = [ Station(self.lib, 8600626, "København H", Position(55673063, 12565796)), Station(self.lib, 10844, "Hovedbanegården, Tivoli", Position(55672838, 12566191)), Station(self.lib, 100240000, "København, Bernstorffsgade (fjernbus)", Position(55673899, 12565112)) ] self.assertEqual(other, stations)
def test_station_library_find_nearby(self): self.assertEqual( self.lib.find_nearby(Position(0, 0), 1000), self.handler.handle(['1.0', 'Stations', 'findNearby'], { 'long': '0', 'lat': 0, 'radius': 1000 }))
def setup_v1_0(self): """ Sets up v1.0 of the API :return: """ handler = self.add_handler('1.0') site_library_handler = handler.add_handler('Stations') site_library_handler.add_function( 'findNearby', lambda name, inp: self.library.find_nearby( Position(int(inp['lat']), int(inp['long'])), int(inp['radius']) )) site_library_handler \ .add_handler('departures') \ .add_function('*', lambda name, inp: self.library.station_from_id(int(name[0])).departures())
def setUp(self): self.query_strategy = StubQueryStrategy() self.lib = StationLibrary(self.query_strategy) self.station = self.lib.find_nearby(Position(0, 0))[0] self.query_strategy.called = []
def test_station_from_id_is_empty(self): self.assertEqual(Station(self.lib, 123, '', Position(0, 0)), self.lib.station_from_id(123))
def test_strategy_find_called_with_right_parameters(self): self.lib.find_nearby(Position(1, 2)) self.assertEqual([('find_nearby', [1, 2, 100, 50])], self.query_strategy.called)
def setUp(self): self.query_strategy = StubQueryStrategy() self.lib = StationLibrary(self.query_strategy) self.station = self.lib.find_nearby(Position(0, 0))[0] self.handler = RESTHandler(self.query_strategy)