1 files changed, 42 insertions, 0 deletions
diff --git a/src/Q10/TrainTimeTableTest.java b/src/Q10/TrainTimeTableTest.java
index 2751519..0217030 100644
--- a/src/Q10/TrainTimeTableTest.java
+++ b/src/Q10/TrainTimeTableTest.java
@@ -42,6 +42,48 @@ public class TrainTimeTableTest extends TestCase {
     assertEquals("23:30", station.getArrival());
     assertEquals("00:29", station.getDeparture());
     assertEquals(3, station.getDay());
+
+    station = this.subject.stationIn("Saskatchewan");
+    assertNotNull(station);
+    assertEquals("Saskatchewan", station.getCity());
+    assertEquals("08:30", station.getArrival());
+    assertEquals("08:55", station.getDeparture());
+    assertEquals(3, station.getDay());
+
+    station = this.subject.stationIn("Winnipeg");
+    assertNotNull(station);
+    assertEquals("Winnipeg", station.getCity());
+    assertEquals("21:15", station.getArrival());
+    assertEquals("23:00", station.getDeparture());
+    assertEquals(3, station.getDay());
+
+    station = this.subject.stationIn("Sioux Lookout");
+    assertNotNull(station);
+    assertEquals("Sioux Lookout", station.getCity());
+    assertEquals("05:32", station.getArrival());
+    assertEquals("06:12", station.getDeparture());
+    assertEquals(4, station.getDay());
+
+    station = this.subject.stationIn("Hornepayne");
+    assertNotNull(station);
+    assertEquals("Hornepayne", station.getCity());
+    assertEquals("16:05", station.getArrival());
+    assertEquals("16:40", station.getDeparture());
+    assertEquals(4, station.getDay());
+
+    station = this.subject.stationIn("Capreol");
+    assertNotNull(station);
+    assertEquals("Capreol", station.getCity());
+    assertEquals("00:48", station.getArrival());
+    assertEquals("01:18", station.getDeparture());
+    assertEquals(5, station.getDay());
+
+    station = this.subject.stationIn("Toronto");
+    assertNotNull(station);
+    assertEquals("Toronto", station.getCity());
+    assertEquals("10:00", station.getArrival());
+    assertEquals("-", station.getDeparture());
+    assertEquals(5, station.getDay());
   }
 
   public void test_stationIn() {