metra_schedule 0.2 → 0.2.1

data/lib/metra/cacher.rb

@@ -6,7 +6,7 @@ module MetraSchedule
     attr_accessor :cache_dir
 
     def initialize
-      @cache_dir = "/home/#{ENV['USER']}/.metra_schedule"
+      @cache_dir = "#{ENV['HOME']}/.metra_schedule"
     end
 
     def self.load_from_cache(line)

data/lib/metra/line.rb

@@ -93,7 +93,7 @@ module MetraSchedule
       unless start_or_destination == :start or start_or_destination == :destination
         raise ArgumentError.new "First argument must be either :start or :destination"
       end
-      raise ArgumentError.new "Not a valid station" unless LINES[:up_nw][:stations].include?(station)
+      raise ArgumentError.new "Not a valid station" unless LINES[@line_key][:stations].include?(station)
       @start = station if start_or_destination == :start
       @destination = station if start_or_destination == :destination 
       self
@@ -116,6 +116,12 @@ module MetraSchedule
       self
     end
 
+    def deduce_direction_by_time
+      @dir = :inbound if Time.now < Time.parse("12:00PM")
+      @dir = :outbound if Time.now > Time.parse("12:00PM")
+      self
+    end
+
     def trains
       return [] unless engines
       engines.find_all do |engine|

data/lib/metra/stop.rb

@@ -14,5 +14,9 @@ module MetraSchedule
       @time > time
     end
 
+    def self.pretty_print(sym)
+      sym.to_s.split("_").map {|s| s.capitalize}.join("\s")
+    end
+
   end
 end

data/lib/metra/train.rb

@@ -22,5 +22,11 @@ module MetraSchedule
       false
     end
 
+    def departure_and_arrival(start, destination)
+      departure = @stops.find {|s| s.station == start}.time
+      arrival = @stops.find {|s| s.station == destination}.time
+      {:departure => departure, :arrival => arrival}
+    end
+
   end
 end

data/test/test_line.rb

@@ -1,5 +1,6 @@
 require 'test/unit'
 require File.join(File.dirname(__FILE__), "../lib", "metra")
+require File.join(File.dirname(__FILE__), "timecop", "lib", "timecop")
 
 class TestLine < Test::Unit::TestCase
 
@@ -250,4 +251,15 @@ class TestLine < Test::Unit::TestCase
     assert_equal([train1, train2, train3, train4], line.trains)
   end
 
+
+  def test_deduce_direction_by_time
+    Timecop.freeze(Time.parse("11:00AM"))
+    line = Metra.new.line(:up_nw).deduce_direction_by_time
+    assert_equal(:inbound, line.dir)
+
+    Timecop.freeze(Time.parse("12:01PM"))
+    line = Metra.new.line(:up_nw).deduce_direction_by_time
+    assert_equal(:outbound, line.dir)
+  end
+
 end

data/test/test_stop.rb

@@ -28,5 +28,9 @@ class TestLine < Test::Unit::TestCase
     assert(stop2.is_after?(Time.parse("12:35")))
   end
 
+  def test_pretty_print
+    assert_equal("Chicago Station", MetraSchedule::Stop.pretty_print(:chicago_station))
+  end
+
 
 end

data/test/test_train.rb

@@ -51,4 +51,15 @@ class TestLine < Test::Unit::TestCase
     assert_equal(false, @@t.in_time?(:barrington, Time.parse("12:35")))
   end
 
+  def test_departure_and_arrival
+    stop1 = MetraSchedule::Stop.new :station => :barrington, :time => Time.parse("12:30")
+    stop2 = MetraSchedule::Stop.new :station => :arlington_heights, :time => Time.parse("12:40")
+    @@t = MetraSchedule::Train.new :train_num => 651, :bike_limit => 12, :schedule => :weekday, :direction => :inbound, :stops => [stop1, stop2]
+
+    l = Metra.new.line(:up_nw)
+    l.engines = [@@t]
+    train = l.trains.first
+    assert_equal({:departure => Time.parse("12:30"), :arrival => Time.parse("12:40")}, train.departure_and_arrival(:barrington, :arlington_heights))
+  end
+
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: metra_schedule
 version: !ruby/object:Gem::Version 
-  version: "0.2"
+  version: 0.2.1
 platform: ruby
 authors: 
 - Blake Smith