daru_lite 0.1

data/spec/date_time/offsets_spec.rb

@@ -0,0 +1,465 @@
+include DaruLite
+
+describe DateOffset do
+  context "#initialize, #+, #-" do
+    it "creates a seconds offset" do
+      offset = DateOffset.new(secs: 5)
+      expect(offset + DateTime.new(2012,3,4,23,4,00)).to eq(
+        DateTime.new(2012,3,4,23,4,05))
+      expect(offset - DateTime.new(2012,4,2,22,4,23)).to eq(
+        DateTime.new(2012,4,2,22,4,18))
+    end
+
+    it "creates a minutes offset"  do
+      offset = DateOffset.new(mins: 2)
+      expect(offset + DateTime.new(2013,4,5,12,45,44)).to eq(
+        DateTime.new(2013,4,5,12,47,44))
+    end
+
+    it "creates an hours offset" do
+      offset = DateOffset.new(hours: 3)
+      expect(offset + DateTime.new(2024,3,2)).to eq(
+        DateTime.new(2024,3,2,03,0,0))
+    end
+
+    it "creates a days offset" do
+      offset = DateOffset.new(days: 12)
+      expect(offset + DateTime.new(2012,5,4)).to eq(
+        DateTime.new(2012,5,16))
+    end
+
+    it "creates a weeks offset" do
+      offset = DateOffset.new(weeks: 2)
+      expect(offset + DateTime.new(2012,3,1)).to eq(
+        DateTime.new(2012,3,15))
+    end
+
+    it "creates a months offset" do
+      offset = DateOffset.new(months: 1)
+      expect(offset + DateTime.new(2012,3,1)).to eq(
+        DateTime.new(2012,4,1))
+    end
+
+    it "creates a years offset" do
+      offset = DateOffset.new(years: 2)
+      expect(offset + DateTime.new(2012,5,30)).to eq(
+        DateTime.new(2014,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,1)).to eq(
+        DateTime.new(2012,3,13))
+    end
+  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 DaruLite::Offsets
+describe Offsets do
+  describe Second do
+    before do
+      @offset = Offsets::Second.new(5)
+    end
+
+    context "#initialize" 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
+    end
+
+    context "#-" do
+      it "reduces by seconds" do
+        expect(@offset - DateTime.new(2012,2,3,12,4,23)).to eq(
+          DateTime.new(2012,2,3,12,4,18))
+      end
+    end
+
+    context "#==" do
+      it 'equals the same offset' do
+        expect(Offsets::Second.new(5)).to eq(Offsets::Second.new(5))
+      end
+
+      it 'does equal a different offset' do
+        expect(Offsets::Second.new(5)).to_not eq(Offsets::Second.new(4))
+      end
+
+      it 'equals offets with the same period' do
+        expect(Offsets::Second.new(60)).to eq(Offsets::Minute.new())
+      end
+    end
+  end
+
+  describe Minute do
+    context "#initialize" do
+      it "creates a minutes offset" do
+        offset = Offsets::Minute.new(2)
+        expect(offset + DateTime.new(2013,4,5,12,45,44)).to eq(
+          DateTime.new(2013,4,5,12,47,44))
+      end
+    end
+
+    context "#-" do
+      it "reduces by minutes" do
+      end
+    end
+  end
+
+  describe Hour do
+    context "#initialize" do
+      it "creates an hours offset" do
+        offset = Offsets::Hour.new(3)
+        expect(offset + DateTime.new(2024,3,2)).to eq(
+          DateTime.new(2024,3,2,03,0,0))
+      end
+    end
+
+    context "#-" do
+      it "reduces by hours" do
+      end
+    end
+  end
+
+  describe Day do
+    context "#initialize" do
+      it "creates a days offset" do
+        offset = Offsets::Day.new(12)
+        expect(offset + DateTime.new(2012,5,4)).to eq(
+          DateTime.new(2012,5,16))
+      end
+    end
+
+    context "#-" do
+      it "reduces by days" do
+      end
+    end
+  end
+
+  describe Week do
+    DAYS_ADVANCE = {
+      sunday:    DateTime.new(2015,7,12),
+      monday:    DateTime.new(2015,7,13),
+      tuesday:   DateTime.new(2015,7,14),
+      wednesday: DateTime.new(2015,7,15),
+      thursday:  DateTime.new(2015,7,16),
+      friday:    DateTime.new(2015,7,17),
+      saturday:  DateTime.new(2015,7,11)
+    }
+
+    DAYS_ADVANCE.each.with_index do |day_date, i|
+      offset = Offsets::Week.new(weekday: i)
+
+      context "#initialize" do
+        date = DateTime.new(2015,7,10)
+
+        it "creates anchored Week offset for #{day_date[0]}" do
+          expect(offset + date).to eq(day_date[1])
+        end
+      end
+
+      context "#on_offset?" do
+        it "checks if given DateTime is on the offset itself? (#{day_date[0]})" do
+          expect(offset.on_offset?(DAYS_ADVANCE[day_date[0]])).to eq(true)
+        end
+      end
+    end
+
+    context "#-" do
+      DAYS_RETREAT = {
+        sunday:    DateTime.new(2015,7,12),
+        monday:    DateTime.new(2015,7,13),
+        tuesday:   DateTime.new(2015,7,14),
+        wednesday: DateTime.new(2015,7,8),
+        thursday:  DateTime.new(2015,7,9),
+        friday:    DateTime.new(2015,7,10),
+        saturday:  DateTime.new(2015,7,11)
+      }
+      date = DateTime.new(2015,7,15)
+
+      DAYS_RETREAT.each.with_index do |day_date, i|
+        it "decreases the date to nearest preceding #{day_date[0]}" do
+          offset = Offsets::Week.new(weekday: i)
+          expect(offset - date).to eq(day_date[1])
+        end
+      end
+    end
+  end
+
+  describe Month do
+    context "#initialize" do
+      it "creates a month offset" do
+        offset = Offsets::Month.new(3)
+        expect(offset + DateTime.new(2012,2,29)).to eq(
+          DateTime.new(2012,5,29))
+      end
+    end
+
+    context "#-" do
+      it "reduces date by a month" do
+        offset = Offsets::Month.new(2)
+        expect(offset - DateTime.new(2012,4,2)).to eq(
+          DateTime.new(2012,2,2))
+      end
+    end
+  end
+
+  describe MonthBegin do
+    before do
+      @offset = Offsets::MonthBegin.new
+      @n_offset = Offsets::MonthBegin.new(3)
+    end
+
+    context "#+" do
+      it "offsets to beginning of next month" do
+        expect(@offset + DateTime.new(2012,3,25)).to eq(
+          DateTime.new(2012,4,1))
+
+        expect(@n_offset + DateTime.new(2011,3,1,5)).to eq(
+          DateTime.new(2011,6,1,5))
+      end
+    end
+
+    context "#on_offset?" 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
+      it "decreases to beginning of the current month if not on offset" do
+        expect(@offset - DateTime.new(2012,4,5)).to eq(
+          DateTime.new(2012,4,1))
+
+        expect(@n_offset - DateTime.new(2012,5,3)).to eq(
+          DateTime.new(2012,3,1))
+      end
+
+      it "decreases to beginning of the previous month if on offset" do
+        expect(@offset - DateTime.new(2012,5,1)).to eq(
+          DateTime.new(2012,4,1))
+
+        expect(@n_offset - DateTime.new(2012,6,1)).to eq(
+          DateTime.new(2012,3,1))
+      end
+    end
+  end
+
+  describe MonthEnd do
+    before do
+      @offset = Offsets::MonthEnd.new
+      @n_offset = Offsets::MonthEnd.new(2)
+    end
+
+    context "#+" do
+      it "increases date to end of next month if on offset" do
+        expect(@offset + DateTime.new(2012,2,29)).to eq(
+          DateTime.new(2012,3,31))
+
+        expect(@n_offset + DateTime.new(2012,2,29)).to eq(
+          DateTime.new(2012,4,30))
+      end
+
+      it "increases date to end of this month if not on offset" do
+        expect(@offset + DateTime.new(2012,4,4)).to eq(
+          DateTime.new(2012,4,30))
+
+        expect(@n_offset + DateTime.new(2012,5,2)).to eq(
+          DateTime.new(2012,6,30))
+      end
+    end
+
+    context "#-" do
+      it "decreases to end of the previous month" do
+        expect(@offset - DateTime.new(2012,2,29)).to eq(
+          DateTime.new(2012,1,31))
+
+        expect(@n_offset - DateTime.new(2015,3,3)).to eq(
+          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
+    before do
+      @offset = Offsets::Year.new
+    end
+
+    context "#+" do
+      it "increaes date by a year" do
+        expect(@offset + DateTime.new(2012,5,2)).to eq(
+          DateTime.new(2013,5,2))
+      end
+    end
+
+    context "#-" do
+      it "decreases date by a year" do
+        expect(@offset - DateTime.new(2011,6,25)).to eq(
+          DateTime.new(2010,6,25))
+      end
+    end
+  end
+
+  describe YearBegin do
+    before do
+      @offset = Offsets::YearBegin.new
+      @n_offset = Offsets::YearBegin.new(2)
+    end
+
+    context "#+" do
+      it "offsets date to future" do
+        expect(@n_offset + DateTime.new(2012,3,25)).to eq(
+          DateTime.new(2014,1,1))
+      end
+    end
+
+    context "#on_offset?" 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
+
+    context "#-" do
+      it "decreases to beginning of the year if not on offset" do
+        expect(@offset - DateTime.new(2012,4,2)).to eq(
+          DateTime.new(2012,1,1))
+
+        expect(@n_offset - DateTime.new(2012,5,5)).to eq(
+          DateTime.new(2011,1,1))
+      end
+
+      it "decreases to beginning of previous year if on offset" do
+        expect(@offset - DateTime.new(2012,1,1)).to eq(
+          DateTime.new(2011,1,1))
+
+        expect(@n_offset - DateTime.new(2013,1,1)).to eq(
+          DateTime.new(2011,1,1))
+      end
+    end
+  end
+
+  describe YearEnd do
+    before do
+      @offset = Offsets::YearEnd.new
+      @n_offset = Offsets::YearEnd.new(2)
+    end
+
+    context "#+" do
+      it "increases to end of same year if not on offset" do
+        expect(@offset + DateTime.new(2011,5,2,4,2)).to eq(
+          DateTime.new(2011,12,31,4,2))
+
+        expect(@n_offset + DateTime.new(2011,5,2,4,2)).to eq(
+          DateTime.new(2012,12,31,4,2))
+      end
+
+      it "increases to end of next year if on offset" do
+        expect(@offset + DateTime.new(2012,12,31,4)).to eq(
+          DateTime.new(2013,12,31,4))
+
+        expect(@n_offset + DateTime.new(2012,12,31,4)).to eq(
+          DateTime.new(2014,12,31,4))
+      end
+    end
+
+    context "#-" do
+      it "decreases to end of previous year" do
+        expect(@offset - DateTime.new(2012,2,3)).to eq(
+          DateTime.new(2011,12,31))
+
+        expect(@n_offset - DateTime.new(2011,5,6)).to eq(
+          DateTime.new(2010,12,31))
+      end
+    end
+
+    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
+end

data/spec/extensions/which_dsl_spec.rb

@@ -0,0 +1,38 @@
+require 'daru_lite/extensions/which_dsl'
+
+describe "which DSL" do
+  before do
+    @df = DaruLite::DataFrame.new({
+      number: [1,2,3,4,5,6,Float::NAN],
+      sym: [:one, :two, :three, :four, :five, :six, :seven],
+      names: ['sameer', 'john', 'james', 'omisha', 'priyanka', 'shravan',nil]
+    })
+  end
+
+  it "accepts simple single eq statement" do
+    answer = DaruLite::DataFrame.new({
+      number: [4],
+      sym: [:four],
+      names: ['omisha']
+      }, index: DaruLite::Index.new([3])
+    )
+    expect( @df.which{ `number` == 4 } ).to eq(answer)
+  end
+
+  it "accepts somewhat complex comparison operator chaining" do
+    answer = DaruLite::DataFrame.new({
+      number: [3,4],
+      sym: [:three, :four],
+      names: ['james', 'omisha']
+    }, index: DaruLite::Index.new([2,3]))
+    expect(
+      @df.which{ (`names` == 'james') | (`sym` == :four) }
+      ).to eq(answer)
+  end
+
+  it "accepts vector methods" do
+    # expect( @df.which{ `number` == `number`.only_valid.max } ).to eq(@df.row_at(5)) # row_at(5) return a Vector object!?
+    expect( @df.which{ `number` <= `number`.only_valid.max } ).to eq(@df.row_at(0..5))
+  end
+
+end