daru 0.1.4.1 → 0.1.5

data/spec/date_time/offsets_spec.rb

@@ -54,6 +54,72 @@ describe DateOffset do
   end
 end
 
+describe NegativeDateOffset do
+  context "#initialize, #+, #-" do
+    it "creates a seconds offset" do
+      offset = -DateOffset.new(secs: 5)
+      expect(offset + DateTime.new(2012,3,4,23,4,05)).to eq(
+        DateTime.new(2012,3,4,23,4,00))
+      expect(offset - DateTime.new(2012,4,2,22,4,18)).to eq(
+        DateTime.new(2012,4,2,22,4,23))
+    end
+
+    it "creates a minutes offset"  do
+      offset = -DateOffset.new(mins: 2)
+      expect(offset + DateTime.new(2013,4,5,12,47,44)).to eq(
+        DateTime.new(2013,4,5,12,45,44))
+    end
+
+    it "creates an hours offset" do
+      offset = -DateOffset.new(hours: 3)
+      expect(offset + DateTime.new(2024,3,2,3)).to eq(
+        DateTime.new(2024,3,2,0))
+    end
+
+    it "creates a days offset" do
+      offset = -DateOffset.new(days: 12)
+      expect(offset + DateTime.new(2012,5,16)).to eq(
+        DateTime.new(2012,5,4))
+    end
+
+    it "creates a weeks offset" do
+      offset = -DateOffset.new(weeks: 2)
+      expect(offset + DateTime.new(2012,3,15)).to eq(
+        DateTime.new(2012,3,1))
+    end
+
+    it "creates a months offset" do
+      offset = -DateOffset.new(months: 1)
+      expect(offset + DateTime.new(2012,4,1)).to eq(
+        DateTime.new(2012,3,1))
+    end
+
+    it "creates a years offset" do
+      offset = -DateOffset.new(years: 2)
+      expect(offset + DateTime.new(2014,5,30)).to eq(
+        DateTime.new(2012,5,30))
+    end
+
+    it "supports 'n' option to apply same offset multiple times" do
+      offset = -DateOffset.new(days: 3, n: 4)
+      expect(offset + DateTime.new(2012,3,13)).to eq(
+        DateTime.new(2012,3,1))
+    end
+  end
+
+  context "#-@" do
+    it "creates a date offset" do
+      negative_offset = -DateOffset.new(secs: 5)
+      offset = -negative_offset
+
+      expect(offset + DateTime.new(2012,3,4,23,4,0)).to eq(
+        DateTime.new(2012,3,4,23,4,5))
+      expect(offset - DateTime.new(2012,4,2,22,4,23)).to eq(
+        DateTime.new(2012,4,2,22,4,18))
+    end
+  end
+end
+
 include Daru::Offsets
 describe Offsets do
   describe Second do
@@ -68,9 +134,6 @@ describe Offsets do
       end
     end
 
-    context "#on_offset?" do
-    end
-
     context "#-" do
       it "reduces by seconds" do
         expect(@offset - DateTime.new(2012,2,3,12,4,23)).to eq(
@@ -212,6 +275,10 @@ describe Offsets do
       it "returns true if date is on the offset" do
         expect(@offset.on_offset?(DateTime.new(2012,4,1))).to eq(true)
       end
+
+      it "returns false if date is not on the offset" do
+        expect(@offset.on_offset?(DateTime.new(2012,4,30))).to eq(false)
+      end
     end
 
     context "#-" do
@@ -266,6 +333,16 @@ describe Offsets do
           DateTime.new(2015,1,31))
       end
     end
+
+    context "#on_offset?" do
+      it "returns true if date is on the offset" do
+        expect(@offset.on_offset?(DateTime.new(2012,4,30))).to eq(true)
+      end
+
+      it "returns false if date is not on the offset" do
+        expect(@offset.on_offset?(DateTime.new(2012,4,1))).to eq(false)
+      end
+    end
   end
 
   describe Year do
@@ -305,6 +382,9 @@ describe Offsets do
       it "checks if date is on the offset" do
         expect(@offset.on_offset?(DateTime.new(2012,1,1))).to eq(
           true)
+
+        expect(@offset.on_offset?(DateTime.new(2012,12,31))).to eq(
+          false)
       end
     end
 
@@ -364,6 +444,7 @@ describe Offsets do
     context "#on_offset?" do
       it "reports whether on offset or not" do
         expect(@offset.on_offset?(DateTime.new(2012,12,31))).to eq(true)
+        expect(@offset.on_offset?(DateTime.new(2012,1,1))).to eq(false)
       end
     end
   end

data/spec/formatters/table_formatter_spec.rb

@@ -74,6 +74,15 @@ describe Daru::Formatters::Table do
     }.unindent}
   end
 
+  context 'with empty data' do
+    let(:data) { [] }
+    let(:headers) { [] }
+    let(:row_headers) { [] }
+
+    it { is_expected.to eq '' }
+  end
+
+
   context '<more> threshold' do
     let(:options) { {threshold: threshold} }
     context 'lower than data size' do

data/spec/index/categorical_index_spec.rb

@@ -142,12 +142,14 @@ describe Daru::CategoricalIndex do
       subject { idx.add :c }
       
       its(:to_a) { is_expected.to eq [:a, 1, :a, 1, :c] }
+      its(:categories) { is_expected.to eq [:a, 1, :c] }
     end
     
     context "multiple indexes" do
       subject { idx.add :c, :d }
       
       its(:to_a) { is_expected.to eq [:a, 1, :a, 1, :c, :d] }
+      its(:categories) { is_expected.to eq [:a, 1, :c, :d] }
     end
   end
   

data/spec/index/index_spec.rb

@@ -54,7 +54,7 @@ describe Daru::Index do
     end
   end
 
-  context '#keys' do
+  context '#key' do
     subject(:idx) { Daru::Index.new ['speaker', 'mic', 'guitar', 'amp'] }
 
     it 'returns key by position' do
@@ -280,4 +280,19 @@ describe Daru::Index do
       it { expect { idx.at 2, 3 }.to raise_error IndexError }
     end
   end
-end
+
+  # This context validate Daru::Index is like an enumerable.
+  # #map and #select are samples and we do not need tests all
+  # enumerable methods.
+  context "Enumerable" do
+    let(:idx) { Daru::Index.new ['speaker', 'mic', 'guitar', 'amp'] }
+
+    context "#map" do
+      it { expect(idx.map(&:upcase)).to eq(['SPEAKER', 'MIC', 'GUITAR', 'AMP']) }
+    end
+
+    context "select" do
+      it { expect(idx.select {|w| w[0] == 'g' }).to eq(['guitar']) }
+    end
+  end
+end

data/spec/{math → maths}/statistics/dataframe_spec.rb

@@ -48,7 +48,7 @@ describe Daru::DataFrame do
 
   context "#mode" do
     it "calculates mode of single level numeric only vectors and returns values in a Vector" do
-      expect(@df.mode).to eq(Daru::Vector.new([2, 4, 20], index: [:d, :e, :f]))
+      expect(@df.mode).to eq(Daru::Vector.new([Daru::Vector.new([2,3]), Daru::Vector.new([4,6]), Daru::Vector.new([20,30])], index: [:d, :e, :f]))
     end
   end
 

data/spec/{math → maths}/statistics/vector_spec.rb

@@ -103,10 +103,15 @@ describe Daru::Vector do
       end
 
       context "#mode" do
-        it "returns the mode" do
+        it "returns the single modal value as a numeric" do
           mode_test_example = Daru::Vector.new [1,2,3,2,4,4,4,4], dtype: dtype
           expect(mode_test_example.mode).to eq(4)
         end
+
+        it "returns multiple modal values as a vector" do
+          mode_test_example = Daru::Vector.new [1,2,2,2,3,2,4,4,4,4], dtype: dtype
+          expect(mode_test_example.mode).to eq(Daru::Vector.new [2,4], dtype: dtype)
+        end
       end
 
       context "#describe" do
@@ -230,18 +235,8 @@ describe Daru::Vector do
   end
 
   context "#frequencies" do
-    it "calculates frequencies" do
-      vector = Daru::Vector.new([5,5,5,5,5,6,6,7,8,9,10,1,2,3,4,nil,-99,-99])
-      expect(vector.frequencies).to eq({
-        1=>1, 2=>1, 3=>1, 4=>1, 5=>5,
-        6=>2, 7=>1, 8=>1, 9=>1,10=>1, -99=>2
-      })
-    end
-  end
-
-  context "#freqs" do
     let(:vector) { Daru::Vector.new([5,5,5,5,5,6,6,7,8,9,10,1,2,3,4,nil,-99,-99]) }
-    subject { vector.freqs }
+    subject { vector.frequencies }
     it { is_expected.to eq Daru::Vector.new(
       [5,2,1,1,1,1,1,1,1,1,2],
       index: [5,6,7,8,9,10,1,2,3,4,-99]

data/spec/plotting/gruff/category_spec.rb

@@ -0,0 +1,44 @@
+describe Daru::Vector, 'plotting category vector with gruff' do
+  before { Daru.plotting_library = :gruff }
+  let(:dv) { Daru::Vector.new [1, 2, 3], type: :category }
+
+  context 'bar' do
+    let(:plot) { instance_double 'Gruff::Bar' }
+    before { allow(Gruff::Bar).to receive(:new).and_return(plot) }
+    it 'plots bar graph' do
+      expect(plot).to receive :labels=
+      expect(plot).to receive :data
+      dv.plot type: :bar
+    end
+    
+    it 'plots bar graph with block' do
+      expect(plot).to receive :labels=
+      expect(plot).to receive :data
+      expect(plot).to receive :title=
+      dv.plot(type: :bar) { |p| p.title = 'hello' }
+    end
+  end
+  
+  context 'pie' do
+    let(:plot) { instance_double 'Gruff::Pie' }
+    before { allow(Gruff::Pie).to receive(:new).and_return(plot) }
+    it 'plots pie graph' do
+      expect(plot).to receive(:data).exactly(3).times
+      dv.plot type: :pie
+    end
+  end
+  
+  context 'sidebar' do
+    let(:plot) { instance_double 'Gruff::SideBar' }
+    before { allow(Gruff::SideBar).to receive(:new).and_return(plot) }
+    it 'plots sidebar graph' do
+      expect(plot).to receive :labels=
+      expect(plot).to receive(:data).exactly(3).times
+      dv.plot type: :sidebar
+    end
+  end
+
+  context 'invalid type' do
+    it { expect { dv.plot type: :lol }.to raise_error ArgumentError }
+  end
+end

data/spec/plotting/gruff/dataframe_spec.rb

@@ -0,0 +1,84 @@
+require 'spec_helper.rb'
+
+describe Daru::DataFrame, 'plotting dataframe using gruff' do
+  before { Daru.plotting_library = :gruff }
+  let(:df) do
+    Daru::DataFrame.new({
+      a: [1, 3, 5, 2, 5, 0],
+      b: [1, 5, 2, 5, 1, 0],
+      c: [1, 6, 7, 2, 6, 0]
+    }, index: 'a'..'f')
+  end
+
+  context 'bar' do
+    let(:plot) { instance_double 'Gruff::Bar' }
+    before { allow(Gruff::Bar).to receive(:new).and_return(plot) }
+    it 'plots bar graph' do
+      expect(plot).to receive :labels=
+      expect(plot).to receive(:data).exactly(3).times
+      df.plot type: :bar
+    end
+    
+    it 'plots bar graph with block' do
+      expect(plot).to receive :labels=
+      expect(plot).to receive(:data).exactly(3).times
+      expect(plot).to receive :title=
+      df.plot(type: :bar) { |p| p.title = 'hello' }
+    end
+
+    it 'plots with specified columns' do
+      expect(plot).to receive :labels=
+      expect(plot).to receive(:data).exactly(2).times
+      df.plot type: :bar, y: [:a, :b]
+    end
+  end
+
+  context 'line' do
+    let(:plot) { instance_double 'Gruff::Line' }
+    before { allow(Gruff::Line).to receive(:new).and_return(plot) }
+    it 'plots line graph' do
+      expect(plot).to receive :labels=
+      expect(plot).to receive(:data).exactly(3).times
+      df.plot type: :line
+    end
+  end
+
+  context 'scatter' do
+    let(:plot) { instance_double 'Gruff::Scatter' }
+    before { allow(Gruff::Scatter).to receive(:new).and_return(plot) }
+    it 'plots scatter graph' do
+      expect(plot).to receive(:data).exactly(3).times
+      df.plot type: :scatter
+    end
+
+    it 'plots with specified columns' do
+      expect(plot).to receive(:data).exactly(1).times
+      df.plot type: :scatter, x: :c, y: :a
+    end
+  end
+  
+  context 'invalid type' do
+    it { expect { df.plot type: :lol }.to raise_error ArgumentError }
+  end
+end
+
+describe Daru::DataFrame, 'dataframe category plotting with gruff' do
+  before { Daru.plotting_library = :gruff }
+  let(:df) do
+    Daru::DataFrame.new({
+      a: [1, 3, 5, 2, 5, 0],
+      b: [1, 5, 2, 5, 1, 0],
+      c: [:a, :b, :a, :a, :b, :a]
+    }, index: 'a'..'f')
+  end
+  before { df.to_category :c }
+  
+  context 'scatter' do
+    let(:plot) { instance_double 'Gruff::Scatter' }
+    before { allow(Gruff::Scatter).to receive(:new).and_return(plot) }    
+    it 'plots scatter plot categorized by category vector' do
+      expect(plot).to receive(:data).exactly(2).times
+      df.plot type: :scatter, x: :a, y: :b, categorized: { by: :c }
+    end
+  end
+end

data/spec/plotting/gruff/vector_spec.rb

@@ -0,0 +1,70 @@
+require 'spec_helper.rb'
+
+describe Daru::Vector, 'plotting vector with gruff' do
+  let(:dv) { Daru::Vector.new [1, 2, 3] }
+  before { Daru.plotting_library = :gruff }
+
+  context 'line' do
+    let(:plot) { instance_double 'Gruff::Line' }
+    before { allow(Gruff::Line).to receive(:new).and_return(plot) }
+    
+    it 'plots line graph without block' do
+      expect(plot).to receive(:labels=)
+      expect(plot).to receive(:data)
+      dv.plot type: :line
+    end
+    
+    it 'plots line graph with block' do
+      expect(plot).to receive :labels=
+      expect(plot).to receive :data
+      expect(plot).to receive :title=
+      dv.plot(type: :line) { |p| p.title = 'hello' }
+    end
+  end
+  
+  context 'bar' do
+    let(:plot) { instance_double 'Gruff::Bar' }
+    before { allow(Gruff::Bar).to receive(:new).and_return(plot) }
+    
+    it 'plots bar graph' do
+      expect(plot).to receive :labels=
+      expect(plot).to receive :data
+      dv.plot type: :bar
+    end
+  end
+  
+  context 'pie' do
+    let(:plot) { instance_double 'Gruff::Pie' }
+    before { allow(Gruff::Pie).to receive(:new).and_return(plot) }
+    
+    it 'plots pie graph' do
+      expect(plot).to receive(:data).exactly(3).times
+      dv.plot type: :pie
+    end
+  end
+  
+  context 'scatter' do
+    let(:plot) { instance_double 'Gruff::Scatter' }
+    before { allow(Gruff::Scatter).to receive(:new).and_return(plot) }
+
+    it 'plots scatter graph' do
+      expect(plot).to receive :data
+      dv.plot type: :scatter
+    end
+  end
+  
+  context 'sidebar' do
+    let(:plot) { instance_double 'Gruff::SideBar' }
+    before { allow(Gruff::SideBar).to receive(:new).and_return(plot) }
+
+    it 'plots sidebar' do
+      expect(plot).to receive :labels=
+      expect(plot).to receive(:data).exactly(3).times
+      dv.plot type: :sidebar
+    end
+  end
+  
+  context 'invalid type' do
+    it { expect { dv.plot type: :lol }.to raise_error ArgumentError }
+  end
+end

data/spec/plotting/nyaplot/category_spec.rb

@@ -0,0 +1,51 @@
+require 'spec_helper.rb'
+
+describe Daru::Vector, 'plotting category' do
+  let(:plot) { instance_double('Nyaplot::Plot') }
+  let(:diagram) { instance_double('Nyaplot::Diagram') }
+  let(:dv) do
+    Daru::Vector.new ['III']*10 + ['II']*5 + ['I']*5,
+      type: :category,
+      categories: ['I', 'II', 'III']
+  end  
+  before do
+    Daru.plotting_library = :nyaplot
+    allow(Nyaplot::Plot).to receive(:new).and_return(plot)
+  end
+  context 'bar' do
+    it 'plots bar graph taking a block' do
+      expect(plot).to receive(:add).with(:bar, ['I', 'II', 'III'], [5, 5, 10])
+      expect(plot).to receive :x_label
+      expect(plot).to receive :y_label
+      expect(plot).to receive(:show)
+      dv.plot(type: :bar) do |p|
+        p.x_label 'Categories'
+        p.y_label 'Frequency'
+      end
+    end
+
+    it 'plots bar graph without taking a block' do
+      expect(plot).to receive(:add).with(:bar, ["I", "II", "III"], [5, 5, 10])
+      expect(plot).to receive(:show)
+      dv.plot(type: :bar)
+    end
+
+    it 'plots bar graph with percentage' do
+      expect(plot).to receive(:add).with(:bar, ["I", "II", "III"], [25, 25, 50])
+      expect(plot).to receive(:yrange).with [0, 100]
+      expect(plot).to receive(:show)
+      dv.plot(type: :bar, method: :percentage)
+    end
+
+    it 'plots bar graph with fraction' do
+      expect(plot).to receive(:add).with(:bar, ["I", "II", "III"], [0.25, 0.25, 0.50])
+      expect(plot).to receive(:yrange).with [0, 1]
+      expect(plot).to receive(:show)
+      dv.plot(type: :bar, method: :fraction)
+    end    
+  end
+
+  context 'other type' do
+    it { expect { dv.plot(type: :scatter) }.to raise_error ArgumentError }
+  end
+end