daru 0.1.3.1 → 0.1.4

data/spec/date_time/data_spec.rb

@@ -1,5 +1,3 @@
-require 'spec_helper'
-
 describe Daru::Vector do
   context "#initialize" do
     it "accepts DateTimeIndex in index option" do
@@ -70,7 +68,7 @@ describe Daru::Vector do
     end
 
     it "assigns multiple elements when index incomplete" do
-      index          = Daru::DateTimeIndex.date_range(:start => '2012', :periods => 100, 
+      index          = Daru::DateTimeIndex.date_range(:start => '2012', :periods => 100,
         :freq => 'MB')
       vector         = Daru::Vector.new([1,2,3,4,5,6,7,8,9,10]*10, index: index)
       vector['2012'] = 666
@@ -88,7 +86,7 @@ describe Daru::DataFrame do
     @a     = [1,2,3,4,5]*20
     @b     = @a.map { |e| e*3 }
     @c     = @a.map(&:to_s)
-    @df    = Daru::DataFrame.new([@a, @b, @c], index: @index, order: @order)    
+    @df    = Daru::DataFrame.new([@a, @b, @c], index: @index, order: @order)
   end
 
   context "#initialize" do
@@ -111,7 +109,7 @@ describe Daru::DataFrame do
 
     it "returns DataFrame when incomplete index" do
       answer = Daru::DataFrame.new(
-        [@a, @c], index: @index, order: Daru::DateTimeIndex.new([ 
+        [@a, @c], index: @index, order: Daru::DateTimeIndex.new([
           DateTime.new(2012,1,3),DateTime.new(2012,3,3)]))
       expect(@df['2012']).to eq(answer)
     end

data/spec/date_time/index_spec.rb

@@ -1,5 +1,3 @@
-require 'spec_helper'
-
 include Daru
 
 describe DateTimeIndex do
@@ -32,6 +30,20 @@ describe DateTimeIndex do
     end
   end
 
+  context '.try_create' do
+    it 'creates index from array of dates' do
+      index = DateTimeIndex.try_create([
+        DateTime.new(2014,7,1),DateTime.new(2014,7,2),
+        DateTime.new(2014,7,3),DateTime.new(2014,7,4)])
+      expect(index.class).to eq(DateTimeIndex)
+    end
+
+    it 'does not creates index from anything else' do
+      index = DateTimeIndex.try_create([:a, :b, :c])
+      expect(index).to be_nil
+    end
+  end
+
   context ".date_range" do
     it "creates DateTimeIndex with default options" do
       index = DateTimeIndex.date_range(:start => DateTime.new(2014,5,3),
@@ -54,6 +66,11 @@ describe DateTimeIndex do
       expect(index.frequency).to eq('D')
     end
 
+    it 'fails on wrong string format' do
+      expect{DateTimeIndex.date_range(start: '2014/5/3', end: '2014/5/10')}
+        .to raise_error(ArgumentError, /Unacceptable date string/)
+    end
+
     it "creates DateTimeIndex of per minute frequency between start and end" do
       index = DateTimeIndex.date_range(start: '2015-7-1',freq: 'M', periods: 10)
 
@@ -88,7 +105,7 @@ describe DateTimeIndex do
     end
 
     it "creates a DateTimeIndex of (sunday) weekly frequency" do
-      index = DateTimeIndex.date_range(start: '2014-8-2', end: '2014-9-8', 
+      index = DateTimeIndex.date_range(start: '2014-8-2', end: '2014-9-8',
         freq: 'W')
 
       expect(index).to eq(DateTimeIndex.new([
@@ -98,7 +115,7 @@ describe DateTimeIndex do
     end
 
     it "creates a DateTimeIndex of (monday) weekly frequency" do
-      index = DateTimeIndex.date_range(:start => '2015-7-6', :periods => 5, 
+      index = DateTimeIndex.date_range(:start => '2015-7-6', :periods => 5,
         :freq => 'W-MON')
       expect(index).to eq(DateTimeIndex.new([
         DateTime.new(2015,7,6), DateTime.new(2015,7,13), DateTime.new(2015,7,20),
@@ -110,7 +127,7 @@ describe DateTimeIndex do
       index = DateTimeIndex.date_range(
         :start => '2017-4-14', :freq => 'MB', :periods => 5)
       expect(index).to eq(DateTimeIndex.new([
-        DateTime.new(2017,5,1), DateTime.new(2017,6,1), 
+        DateTime.new(2017,5,1), DateTime.new(2017,6,1),
         DateTime.new(2017,7,1), DateTime.new(2017,8,1),DateTime.new(2017,9,1)]))
     end
 
@@ -132,13 +149,13 @@ describe DateTimeIndex do
       index = DateTimeIndex.date_range(start: '2014-9',end: '2018-1',freq: 'YE')
 
       expect(index).to eq(DateTimeIndex.new([
-        DateTime.new(2014,12,31), DateTime.new(2015,12,31),DateTime.new(2016,12,31), 
+        DateTime.new(2014,12,31), DateTime.new(2015,12,31),DateTime.new(2016,12,31),
         DateTime.new(2017,12,31)]))
       expect(index.frequency).to eq('YE')
     end
 
     it "creates only specfied number of periods taking precendence over end" do
-      index = DateTimeIndex.date_range(start: '2014-5-5', end: '2015-3', 
+      index = DateTimeIndex.date_range(start: '2014-5-5', end: '2015-3',
         periods: 5, freq: 'YB')
       expect(index).to eq(DateTimeIndex.new([
         DateTime.new(2015,1,1),DateTime.new(2016,1,1),DateTime.new(2017,1,1),
@@ -148,19 +165,45 @@ describe DateTimeIndex do
     it "does not increment start date if it satisifies the anchor" do
       index = DateTimeIndex.date_range(:start => '2012-1-1', freq: 'MB', periods: 4)
       expect(index).to eq(DateTimeIndex.new(
-        [DateTime.new(2012,1,1), DateTime.new(2012,2,1), 
+        [DateTime.new(2012,1,1), DateTime.new(2012,2,1),
          DateTime.new(2012,3,1), DateTime.new(2012,4,1)]))
     end
 
     it "raises error for different start and end timezones" do
       expect {
         DateTimeIndex.date_range(
-          :start => DateTime.new(2012,3,4,12,5,4,"+5:30"), 
+          :start => DateTime.new(2012,3,4,12,5,4,"+5:30"),
           :end => DateTime.new(2013,3,4,12,5,4,"+7:30"), freq: 'M')
       }.to raise_error(ArgumentError)
     end
   end
 
+  context '#inspect' do
+    subject { index.inspect }
+
+    context 'with known frequency' do
+      let(:index){
+        DateTimeIndex.new([
+          DateTime.new(2014,7,1),DateTime.new(2014,7,2),DateTime.new(2014,7,3),
+          DateTime.new(2014,7,4)], freq: :infer)
+      }
+      it { is_expected.to eq \
+        "#<Daru::DateTimeIndex(4, frequency=D) 2014-07-01T00:00:00+00:00...2014-07-04T00:00:00+00:00>"
+      }
+    end
+
+    context 'with unknown frequency' do
+      let(:index){
+        DateTimeIndex.new([
+          DateTime.new(2014,7,1),DateTime.new(2014,7,2),DateTime.new(2014,7,3),
+          DateTime.new(2014,7,4)])
+      }
+      it { is_expected.to eq \
+        "#<Daru::DateTimeIndex(4) 2014-07-01T00:00:00+00:00...2014-07-04T00:00:00+00:00>"
+      }
+    end
+  end
+
   context "#frequency" do
     it "reports the frequency of when a period index is specified" do
       index = DateTimeIndex.new([
@@ -180,7 +223,7 @@ describe DateTimeIndex do
   context "#[]" do
     it "accepts complete time as a string" do
       index = DateTimeIndex.new([
-        DateTime.new(2014,3,3),DateTime.new(2014,3,4),DateTime.new(2014,3,5),DateTime.new(2014,3,6)], 
+        DateTime.new(2014,3,3),DateTime.new(2014,3,4),DateTime.new(2014,3,5),DateTime.new(2014,3,6)],
         freq: :infer)
       expect(index.frequency).to eq('D')
       expect(index['2014-3-5']).to eq(2)
@@ -188,7 +231,7 @@ describe DateTimeIndex do
 
     it "accepts complete time as a DateTime object" do
       index = DateTimeIndex.new([
-        DateTime.new(2014,3,3),DateTime.new(2014,3,4),DateTime.new(2014,3,5),DateTime.new(2014,3,6)], 
+        DateTime.new(2014,3,3),DateTime.new(2014,3,4),DateTime.new(2014,3,5),DateTime.new(2014,3,6)],
         freq: :infer)
       expect(index[DateTime.new(2014,3,6)]).to eq(3)
     end
@@ -202,7 +245,7 @@ describe DateTimeIndex do
     end
 
     it "accepts only year for frequency data" do
-      index = DateTimeIndex.date_range(:start => DateTime.new(2012,3,2), 
+      index = DateTimeIndex.date_range(:start => DateTime.new(2012,3,2),
         periods: 1000, freq: '5D')
       expect(index['2012']).to eq(DateTimeIndex.date_range(
         :start => DateTime.new(2012,3,2), :end => DateTime.new(2012,12,27), freq: '5D'))
@@ -233,10 +276,10 @@ describe DateTimeIndex do
     end
 
     it "accepts year, month, date for frequency data" do
-      index = DateTimeIndex.date_range(:start => DateTime.new(2012,2,29), 
+      index = DateTimeIndex.date_range(:start => DateTime.new(2012,2,29),
         periods: 1000, freq: 'M')
       expect(index['2012-2-29']).to eq(DateTimeIndex.date_range(
-        :start => DateTime.new(2012,2,29), 
+        :start => DateTime.new(2012,2,29),
         :end   => DateTime.new(2012,2,29,16,39,00), freq: 'M'))
     end
 
@@ -291,7 +334,7 @@ describe DateTimeIndex do
     end
 
     it "returns slice upto last element if overshoot in range" do
-      index = DateTimeIndex.date_range(:start => '2012-2-2', :periods => 50, 
+      index = DateTimeIndex.date_range(:start => '2012-2-2', :periods => 50,
         freq: 'M')
       expect(index['2012'..'2013']).to eq(DateTimeIndex.date_range(
         :start => '2012-2-2',:periods => 50, freq: 'M'))
@@ -332,10 +375,75 @@ describe DateTimeIndex do
       }.to raise_error(ArgumentError)
     end
   end
+  
+  context "#pos" do
+    let(:idx) do
+      described_class.new([
+        DateTime.new(2014,3,3),
+        DateTime.new(2014,3,4),
+        DateTime.new(2014,3,5),
+        DateTime.new(2014,3,6)
+        ], freq: :infer
+      )
+    end
+    
+    context "single index" do
+      it { expect(idx.pos '2014-3-4').to eq 1 }
+    end
+    
+    context "multiple indexes" do
+      subject { idx.pos '2014' }
+      
+      it { is_expected.to be_a Array }
+      its(:size) { is_expected.to eq 4 }
+      it { is_expected.to eq [0, 1, 2, 3] }
+    end
+    
+    context "single positional index" do
+      it { expect(idx.pos 1).to eq 1 }
+    end
+    
+    context "multiple positional indexes" do
+      subject { idx.pos 0, 2 }
+      
+      it { is_expected.to be_a Array }
+      its(:size) { is_expected.to eq 3 }
+      it { is_expected.to eq [0, 1, 2] }
+    end
+  end
+  
+  context "#subset" do
+    let(:idx) do
+      described_class.new([
+        DateTime.new(2014,3,3),
+        DateTime.new(2014,3,4),
+        DateTime.new(2014,3,5),
+        DateTime.new(2014,3,6)
+        ], freq: :infer
+      )
+    end
+    
+    context "multiple indexes" do
+      subject { idx.subset '2014' }
+      
+      it { is_expected.to be_a described_class }
+      its(:size) { is_expected.to eq 4 }
+      it { is_expected.to eq idx }
+    end
+    
+    context "multiple positional indexes" do
+      subject { idx.subset 0, 2 }
+      
+      it { is_expected.to be_a described_class }
+      its(:size) { is_expected.to eq 3 }
+      its(:to_a) { is_expected.to eq [DateTime.new(2014, 3, 3),
+        DateTime.new(2014, 3, 4), DateTime.new(2014, 3, 5)] }
+    end
+  end
 
   context "#slice" do
     it "supports both DateTime objects" do
-      index = DateTimeIndex.date_range(:start => '2012', :periods => 50, 
+      index = DateTimeIndex.date_range(:start => '2012', :periods => 50,
         :freq => 'M')
       expect(index.slice(DateTime.new(2012,1,1), DateTime.new(2012,1,1,0,6))).to eq(
         DateTimeIndex.date_range(:start => '2012', :periods => 7, :freq => 'M'))
@@ -352,6 +460,16 @@ describe DateTimeIndex do
       expect(index.slice(DateTime.new(2012), '2013')).to eq(
         DateTimeIndex.date_range(:start => '2012', :periods => 24, :freq => 'MONTH'))
     end
+
+    it "supports two strings" do
+      index = DateTimeIndex.date_range(:start => '2012', :periods => 40, :freq => 'MONTH')
+      expect(index.slice('2012', '2013')).to eq(
+        DateTimeIndex.date_range(:start => '2012', :periods => 24, :freq => 'MONTH'))
+
+      # FIXME: It works this way now, yet I'm faithfully not sure that is most
+      # reasonable behavior. At least MY expectation is "slice(2012, 2013)" returns
+      # "from start of 2012 to start of 2013"... Or am I missing something?.. - zverok
+    end
   end
 
   context "#size" do
@@ -360,6 +478,25 @@ describe DateTimeIndex do
       expect(index.size).to eq(100)
     end
   end
+  
+  context "#add" do
+    before { skip }
+    let(:idx) { Daru::Index.new [:a, :b, :c] }
+
+    context "single index" do
+      subject { idx }
+      before { idx.add :d }
+      
+      its(:to_a) { is_expected.to eq [:a, :b, :c, :d] }
+    end
+    
+    context "mulitple indexes" do
+      subject { idx }
+      before { idx.add :d, :e }
+      
+      its(:to_a) { is_expected.to eq [:a, :b, :c, :d, :e] }
+    end
+  end
 
   context "#to_a" do
     it "returns an Array of ruby Time objects" do
@@ -430,4 +567,4 @@ describe DateTimeIndex do
       expect(index.include?(DateTime.new(2011,4,2))).to eq(false)
     end
   end
-end
+end

data/spec/date_time/offsets_spec.rb

@@ -1,4 +1,3 @@
-require 'spec_helper'
 include Daru
 
 describe DateOffset do
@@ -63,7 +62,7 @@ describe Offsets do
     end
 
     context "#initialize" do
-      it "creates a seconds offset" do    
+      it "creates a seconds offset" do
         expect(@offset + DateTime.new(2012,3,4,23,4,00)).to eq(
           DateTime.new(2012,3,4,23,4,05))
       end
@@ -200,7 +199,7 @@ describe Offsets do
     end
 
     context "#+" do
-      it "offsets to beginning of next month" do 
+      it "offsets to beginning of next month" do
         expect(@offset + DateTime.new(2012,3,25)).to eq(
           DateTime.new(2012,4,1))
 
@@ -296,7 +295,7 @@ describe Offsets do
     end
 
     context "#+" do
-      it "offsets date to future" do 
+      it "offsets date to future" do
         expect(@n_offset + DateTime.new(2012,3,25)).to eq(
           DateTime.new(2014,1,1))
       end

data/spec/fixtures/empties.dat

@@ -0,0 +1,2 @@
+   1    2    3
+   4         6

data/spec/fixtures/strings.dat

@@ -0,0 +1,2 @@
+  test  1 2
+  foo   3 4

data/spec/formatters/table_formatter_spec.rb

@@ -0,0 +1,99 @@
+describe Daru::Formatters::Table do
+  let(:options) { {} }
+  subject {
+    Daru::Formatters::Table
+      .format(data, options.merge(headers: headers, row_headers: row_headers))
+  }
+
+  let(:data) { [[1,2,3], [4,5,6], [7,8,9]] }
+  let(:headers) { [:col1, :col2, :col3] }
+  let(:row_headers) { [:row1, :row2, :row3] }
+
+  context 'simple table' do
+    it { is_expected.to eq %Q{
+      |      col1 col2 col3
+      | row1    1    2    3
+      | row2    4    5    6
+      | row3    7    8    9
+    }.unindent}
+  end
+
+  context 'with nils' do
+    let(:data) { [[1,nil,3], [4,5,nil], [7,8,9]] }
+    let(:headers) { [:col1, :col2, nil] }
+    let(:row_headers) { [:row1, nil, :row3] }
+
+    it { is_expected.to eq %Q{
+      |      col1 col2     |
+      | row1    1  nil    3|
+      |         4    5  nil|
+      | row3    7    8    9|
+    }.unindent}
+  end
+
+  context 'with multivalue row headers' do
+    let(:row_headers) { [[:row,1], [:row,2], [:row,3]] }
+    it { is_expected.to eq %Q{
+      |           col1 col2 col3
+      |  row    1    1    2    3
+      |  row    2    4    5    6
+      |  row    3    7    8    9
+    }.unindent}
+  end
+
+  context 'with multivalue column headers' do
+    let(:headers) { [[:col,1], [:col,2], [:col,3]] }
+  end
+
+  context 'rows only' do
+    let(:data) { [] }
+    let(:headers) { nil }
+    it { is_expected.to eq %Q{
+      | row1
+      | row2
+      | row3
+    }.unindent}
+  end
+
+  context 'columns only' do
+    let(:data) { [] }
+    let(:row_headers) { nil }
+    it { is_expected.to eq %Q{
+      | col1 col2 col3
+    }.unindent}
+  end
+
+  context 'wide values' do
+    let(:options) { {spacing: 2} }
+
+    it { is_expected.to eq %Q{
+      |    co co co
+      | ro  1  2  3
+      | ro  4  5  6
+      | ro  7  8  9
+    }.unindent}
+  end
+
+  context '<more> threshold' do
+    let(:options) { {threshold: threshold} }
+    context 'lower than data size' do
+      let(:threshold) { 2 }
+      it { is_expected.to eq %Q{
+        |      col1 col2 col3
+        | row1    1    2    3
+        | row2    4    5    6
+        |  ...  ...  ...  ...
+      }.unindent}
+    end
+
+    context 'greater than data size' do
+      let(:threshold) { 5 }
+      it { is_expected.to eq %Q{
+        |      col1 col2 col3
+        | row1    1    2    3
+        | row2    4    5    6
+        | row3    7    8    9
+      }.unindent}
+    end
+  end
+end