Upgrade origin-src to google transit feed 1.2.6
[bus.git] / origin-src / transitfeed-1.2.6 / test / testkmlparser.py
blob:a/origin-src/transitfeed-1.2.6/test/testkmlparser.py -> blob:b/origin-src/transitfeed-1.2.6/test/testkmlparser.py
--- a/origin-src/transitfeed-1.2.6/test/testkmlparser.py
+++ b/origin-src/transitfeed-1.2.6/test/testkmlparser.py
@@ -1,1 +1,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()
+