1 files changed, 10 insertions, 0 deletions
diff --git a/src/Q10/TrainTimeTableTest.java b/src/Q10/TrainTimeTableTest.java
index c0ab35a..2751519 100644
--- a/src/Q10/TrainTimeTableTest.java
+++ b/src/Q10/TrainTimeTableTest.java
@@ -43,4 +43,14 @@ public class TrainTimeTableTest extends TestCase {
     assertEquals("00:29", station.getDeparture());
     assertEquals(3, station.getDay());
   }
+
+  public void test_stationIn() {
+    assertNotNull(this.subject.stationIn("Edmonton"));
+    assertEquals("Edmonton", this.subject.stationIn("Edmonton").getCity());
+  }
+
+  public void test_stationIn_lowercase() {
+    assertNotNull(this.subject.stationIn("edmonton"));
+    assertEquals("Edmonton", this.subject.stationIn("edmonton").getCity());
+  }
 }