1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 | #!/usr/bin/python2.5 # Copyright (C) 2007 Google Inc. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. # You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, software # distributed under the License is distributed on an "AS IS" BASIS, # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. # Unit tests for the kmlparser module. import kmlparser import os.path import shutil from StringIO import StringIO import transitfeed import unittest import util class TestStopsParsing(util.GetPathTestCase): 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 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) class FullTests(util.TempDirTestCaseBase): def testNormalRun(self): shutil.copyfile(self.GetTestDataPath('one_stop.kml'), 'one_stop.kml') (out, err) = self.CheckCallWithPath( [self.GetPath('kmlparser.py'), 'one_stop.kml', 'one_stop.zip']) # There will be lots of problems, but ignore them accumulator = util.RecordingProblemAccumulator(self) problems = transitfeed.ProblemReporter(accumulator) schedule = transitfeed.Loader('one_stop.zip', problems=problems).Load() self.assertEquals(len(schedule.GetStopList()), 1) self.assertFalse(os.path.exists('transitfeedcrash.txt')) def testCommandLineError(self): (out, err) = self.CheckCallWithPath([self.GetPath('kmlparser.py')], expected_retcode=2) self.assertMatchesRegex(r'did not provide .+ arguments', err) self.assertMatchesRegex(r'[Uu]sage:', err) self.assertFalse(os.path.exists('transitfeedcrash.txt')) def testCrashHandler(self): (out, err) = self.CheckCallWithPath( [self.GetPath('kmlparser.py'), 'IWantMyCrash', 'output.zip'], stdin_str="\n", expected_retcode=127) self.assertMatchesRegex(r'Yikes', out) crashout = open('transitfeedcrash.txt').read() self.assertMatchesRegex(r'For testCrashHandler', crashout) if __name__ == '__main__': unittest.main() |