Adding validation for uniqueness of service_id X date in calendar_dates

python/test/testtransitfeed.py

 #!/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
 #
@@ skipping 3192 matching lines @@
     self.assertEqual("Bogus Stop (Demo)", e.stop_name)
     self.assertEqual("BOGUS", e.stop_id)
     self.problems.AssertNoMoreExceptions()
     
 
 
 class OnlyCalendarDatesTestCase(LoadTestCase):
   def runTest(self):
     self.Load('only_calendar_dates'),
     self.problems.AssertNoMoreExceptions()
+
+
+class DuplicateServiceIdDateWarningTestCase(MemoryZipTestCase):
+  def runTest(self):
+    # Two lines with the same value of service_id and date.
+    # Test for the warning.
+    self.zip.writestr(
+        'calendar_dates.txt',
+        'service_id,date,exception_type\n'
+        'FULLW,20100604,1\n'
+        'FULLW,20100604,2\n')
+    schedule = self.loader.Load()
+    e = self.problems.PopException('DuplicateID')
+    self.assertEquals('(service_id, date)', e.column_name)
+    self.assertEquals('(FULLW, 20100604)', e.value)
 
 
 class AddStopTimeParametersTestCase(unittest.TestCase):
   def runTest(self):
     problem_reporter = TestFailureProblemReporter(self)
     schedule = transitfeed.Schedule(problem_reporter=problem_reporter)
     route = schedule.AddRoute(short_name="10", long_name="", route_type="Bus")
     stop = schedule.AddStop(40, -128, "My stop")
     # Stop must be added to schedule so that the call
     # AddStopTime -> AddStopTimeObject -> GetStopTimes -> GetStop can work
@@ skipping 1158 matching lines @@
     stoptimes = self.trip1.GetStopTimes()
     for stoptimes_clone in stoptimes_list:
       self.assertEqual(len(stoptimes_clone), len(stoptimes))
       for st_clone, st in zip(stoptimes_clone, stoptimes):
         for name in st.__slots__:
           if name not in ('arrival_secs', 'departure_secs'):
             self.assertEqual(getattr(st, name), getattr(st_clone, name))
 
 if __name__ == '__main__':
   unittest.main()