def testSingleStop(self):
feed = transitfeed.Schedule()
kmlFile = self.GetTestDataPath('one_stop.kml')
kmlparser.KmlParser().Parse(kmlFile, feed)
stops = feed.GetStopList()
self.assertEqual(1, len(stops))
stop = stops[0]
self.assertEqual(u'Stop Name', stop.stop_name)
self.assertAlmostEqual(-93.239037, stop.stop_lon)
self.assertAlmostEqual(44.854164, stop.stop_lat)
write_output = StringIO()
feed.WriteGoogleTransitFeed(write_output)
def runTest(self):
gtfs_input = DataPath('good_feed.zip')
feed1 = transitfeed.Loader(gtfs_input).Load()
kmlwriter.KMLWriter().Write(feed1, self.kml_output)
feed2 = transitfeed.Schedule()
kmlparser.KmlParser().Parse(self.kml_output, feed2)
stop_name_mapper = lambda x: x.stop_name
stops1 = set(map(stop_name_mapper, feed1.GetStopList()))
stops2 = set(map(stop_name_mapper, feed2.GetStopList()))
self.assertEqual(stops1, stops2)
def testSingleShape(self):
feed = transitfeed.Schedule()
kmlFile = self.GetTestDataPath('one_line.kml')
kmlparser.KmlParser().Parse(kmlFile, feed)
shapes = feed.GetShapeList()
self.assertEqual(1, len(shapes))
shape = shapes[0]
self.assertEqual(3, len(shape.points))
self.assertAlmostEqual(44.854240, shape.points[0][0])
self.assertAlmostEqual(-93.238861, shape.points[0][1])
self.assertAlmostEqual(44.853081, shape.points[1][0])
self.assertAlmostEqual(-93.238708, shape.points[1][1])
self.assertAlmostEqual(44.852638, shape.points[2][0])
self.assertAlmostEqual(-93.237923, shape.points[2][1])
write_output = StringIO()
feed.WriteGoogleTransitFeed(write_output)