lib/sqa/stock.rb in sqa-0.0.6 vs lib/sqa/stock.rb in sqa-0.0.7

- old
+ new

@@ -4,13 +4,15 @@ attr_accessor :company_name attr_accessor :df # The DataFrane attr_accessor :ticker def initialize(ticker:, source: :yahoo_finance, type: :csv) - @ticker = ticker + @ticker = ticker.downcase @company_name = "Company Name" klass = "SQA::DataFrame::#{source.to_s.camelize}".constantize - @df = klass.send("from_#{type.downcase}", ticker) + filename = "#{@ticker}.#{type}" + @df = klass.send(:load, filename) + @df[:ticker] = ticker end def to_s "#{ticker} with #{@df.size} data points from #{@df.timestamp.first} to #{@df.timestamp.last}" end