plotrb 0.0.1

data/spec/plotrb/marks_spec.rb

@@ -0,0 +1,46 @@
+require_relative '../spec_helper'
+
+describe 'Mark' do
+
+  subject { ::Plotrb::Mark.rect }
+
+  describe '#initialize' do
+
+    context 'when type is group' do
+
+      subject { ::Plotrb::Mark.group }
+
+      it 'has additional scales, axes, and marks attribute' do
+        subject.attributes.should include(:scales, :axes, :marks)
+      end
+
+    end
+
+  end
+
+  describe 'properties' do
+
+    it 'allows multiple properties' do
+      ::Plotrb::Kernel.stub(:find_data).with('some_data').and_return(true)
+      subject.from('some_data')
+      subject.enter
+      subject.exit
+      subject.properties.keys.should match_array([:enter, :exit])
+    end
+
+  end
+
+  describe '#from' do
+
+    it 'recognizes Data and Transform objects' do
+      foo = ::Plotrb::Transform.facet
+      bar = ::Plotrb::Transform.filter
+      ::Plotrb::Kernel.stub(:find_data).with('some_data').and_return(true)
+      subject.from('some_data', foo, bar)
+      subject.send(:process_from)
+      subject.from.should == {:data => 'some_data', :transform => [foo, bar]}
+    end
+
+  end
+
+end

data/spec/plotrb/scales_spec.rb

@@ -0,0 +1,187 @@
+require_relative '../spec_helper'
+
+describe 'Scale' do
+
+  subject { ::Plotrb::Scale.new }
+
+  describe '#name' do
+
+    it 'sets name of the scale' do
+      subject.name 'foo_scale'
+      subject.name.should == 'foo_scale'
+    end
+
+    it 'raises error if the name is not unique' do
+      subject.name 'foo'
+      ::Plotrb::Kernel.stub(:duplicate_scale?).and_return(true)
+      expect { subject.send(:process_name) }.to raise_error(ArgumentError)
+    end
+
+  end
+
+  describe '#type' do
+
+    it 'does not allow changing type once initialized' do
+      expect { subject.type(:other_type) }.to raise_error(ArgumentError)
+    end
+
+  end
+
+  describe '#domain' do
+
+    context 'when domain is a string reference to a data source' do
+
+      before(:each) do
+        ::Plotrb::Kernel.stub(:find_data).and_return(::Plotrb::Data.new)
+      end
+
+      it 'separates data source and data field' do
+        subject.from('some_data.some_field')
+        ::Plotrb::Data.any_instance.stub(:extra_fields).and_return([])
+        subject.send(:process_domain)
+        subject.domain.data.should == 'some_data'
+        subject.domain.field.should == 'data.some_field'
+      end
+
+      it 'defaults field to index if not provided' do
+        subject.from('some_data')
+        ::Plotrb::Data.any_instance.stub(:extra_fields).and_return([])
+        subject.send(:process_domain)
+        subject.domain.data.should == 'some_data'
+        subject.domain.field.should == 'index'
+      end
+
+      it 'deals with index field properly' do
+        subject.from('some_data.index')
+        ::Plotrb::Data.any_instance.stub(:extra_fields).and_return([])
+        subject.send(:process_domain)
+        subject.domain.data.should == 'some_data'
+        subject.domain.field.should == 'index'
+      end
+
+      it 'recognizes extra fields added by transform' do
+        subject.from('some_data.field')
+        ::Plotrb::Data.any_instance.stub(:extra_fields).and_return([:field])
+        subject.send(:process_domain)
+        subject.domain.data.should == 'some_data'
+        subject.domain.field.should == 'field'
+      end
+    end
+
+    context 'when domain is actual ordinal/categorical data' do
+
+      it 'sets domain directly' do
+        data_set = %w(foo bar baz qux)
+        subject.from(data_set)
+        subject.domain.should == data_set
+      end
+
+    end
+
+    context 'when domain is a quantitative range' do
+
+      it 'sets domain as a two-element array' do
+        subject.from([1,100])
+        subject.domain.should == [1,100]
+      end
+
+    end
+
+  end
+
+  describe '#range' do
+
+    context 'when range is numeric' do
+
+      it 'sets range as a two-element array' do
+        subject.to([1,100])
+        subject.send(:process_range)
+        subject.range.should == [1,100]
+      end
+
+    end
+
+    context 'when range is ordinal' do
+
+      it 'sets range directly' do
+        range_set = %w(foo bar baz qux)
+        subject.to(range_set)
+        subject.send(:process_range)
+        subject.range.should == range_set
+      end
+
+    end
+
+    context 'when range is special literal' do
+
+      it 'sets correct range literal' do
+        subject.to_more_colors
+        subject.send(:process_range)
+        subject.range.should == :category20
+      end
+
+      it 'does not set invalid range literal' do
+        expect { subject.to_foo_bar_range }.to raise_error(NoMethodError)
+        subject.range.should be_nil
+      end
+
+    end
+
+  end
+
+  describe '#exponent' do
+
+    it 'sets the exponent of scale transformation' do
+      subject.in_exponent(10)
+      subject.exponent.should == 10
+    end
+
+  end
+
+  describe '#nice' do
+
+    context 'when scale is time or utc' do
+
+      subject { ::Plotrb::Scale.time }
+
+      it 'sets valid nice literal' do
+        subject.in_seconds
+        subject.nice.should == :second
+      end
+
+      it 'does not set invalid nice literal' do
+        expect { subject.in_millennium }.to raise_error(NoMethodError)
+        subject.nice.should be_nil
+      end
+
+    end
+
+    context 'when scale is quantitative' do
+
+      subject { ::Plotrb::Scale.linear }
+
+      it 'sets nice to true' do
+        subject.nicely
+        subject.nice?.should be_true
+      end
+
+    end
+
+  end
+
+  describe '#method_missing' do
+
+    it 'calls nice if in_foo is called' do
+      subject.type = :time
+      subject.should_receive(:nice).with(:second)
+      subject.in_seconds
+    end
+
+    it 'calls range if to_foo is called' do
+      subject.should_receive(:range).with(:colors)
+      subject.to_colors
+    end
+
+  end
+
+end

data/spec/plotrb/simple_spec.rb

@@ -0,0 +1,61 @@
+#--
+# simple_spec.rb: specs for simple plot shortcuts 
+# Copyright (c) 2013 Colin J. Fuller and the Ruby Science Foundation
+# All rights reserved.
+#
+# Redistribution and use in source and binary forms, with or without
+# modification, are permitted provided that the following conditions
+# are met:
+#   - Redistributions of source code must retain the above copyright
+#   notice, this list of conditions and the following disclaimer.
+#
+#   - Redistributions in binary form must reproduce the above copyright
+#   notice, this list of conditions and the following disclaimer in the
+#   documentation and/or other materials provided with the
+#   distribution.
+#
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
+# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
+# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
+# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
+# HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
+# INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
+# BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS
+# OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED
+# AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
+# LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY
+# WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
+# POSSIBILITY OF SUCH DAMAGE.
+#++
+
+require 'spec_helper'
+require 'plotrb/simple'
+
+describe Plotrb::Simple do 
+
+  context "scatter plots" do
+    before :each do
+      @xdata = [0,1,2,3,4,5]
+      @ydata = [[0,1,4,9,16,25],[0,1,8,27,64,125]]
+    end
+
+    it "should correctly make a scatter plot with supplied options" do
+      Plotrb::Simple.scatter(@xdata, @ydata, markersize: 40, domain: [0,10], range: [0,125], color: ['red', 'blue'], width: 1000, height: 1000, symbol: ['triangle-up', 'triangle-down']).generate_spec.should eq '{"width":1000,"height":1000,"data":[{"name":"scatter","values":[{"x0":0,"y0":0,"y1":0},{"x0":1,"y0":1,"y1":1},{"x0":2,"y0":4,"y1":8},{"x0":3,"y0":9,"y1":27},{"x0":4,"y0":16,"y1":64},{"x0":5,"y0":25,"y1":125}]}],"scales":[{"name":"scatter_x","type":"linear","domain":[0,10],"range":"width"},{"name":"scatter_y","type":"linear","domain":[0,125],"range":"height"}],"marks":[{"type":"symbol","from":{"data":"scatter"},"properties":{"enter":{"size":{"value":40},"shape":{"value":"triangle-up"},"x":{"field":"data.x0","scale":"scatter_x"},"y":{"field":"data.y0","scale":"scatter_y"},"fill":{"value":"red"}}}},{"type":"symbol","from":{"data":"scatter"},"properties":{"enter":{"size":{"value":40},"shape":{"value":"triangle-down"},"x":{"field":"data.x0","scale":"scatter_x"},"y":{"field":"data.y1","scale":"scatter_y"},"fill":{"value":"blue"}}}}],"axes":[{"type":"x","scale":"scatter_x"},{"type":"y","scale":"scatter_y"}]}'
+    end
+
+    it "should correctly make a scatter plot with default options and multiple data series" do
+      Plotrb::Simple.scatter(@xdata, @ydata).generate_spec.should eq '{"width":640,"height":480,"data":[{"name":"scatter","values":[{"x0":0,"y0":0,"y1":0},{"x0":1,"y0":1,"y1":1},{"x0":2,"y0":4,"y1":8},{"x0":3,"y0":9,"y1":27},{"x0":4,"y0":16,"y1":64},{"x0":5,"y0":25,"y1":125}]}],"scales":[{"name":"scatter_x","type":"linear","domain":{"data":"scatter","field":"data.x0"},"range":"width"},{"name":"scatter_y","type":"linear","domain":{"data":"scatter","field":"data.y0"},"range":"height"}],"marks":[{"type":"symbol","from":{"data":"scatter"},"properties":{"enter":{"size":{"value":20},"shape":{"value":"circle"},"x":{"field":"data.x0","scale":"scatter_x"},"y":{"field":"data.y0","scale":"scatter_y"},"fill":{"value":"blue"}}}},{"type":"symbol","from":{"data":"scatter"},"properties":{"enter":{"size":{"value":20},"shape":{"value":"circle"},"x":{"field":"data.x0","scale":"scatter_x"},"y":{"field":"data.y1","scale":"scatter_y"},"fill":{"value":"blue"}}}}],"axes":[{"type":"x","scale":"scatter_x"},{"type":"y","scale":"scatter_y"}]}'
+    end
+
+    it "should correctly make a scatter plot with default options and a single data series" do
+      Plotrb::Simple.scatter(@xdata, [0,1,4,9,16,25]).generate_spec.should eq '{"width":640,"height":480,"data":[{"name":"scatter","values":[{"x0":0,"y0":0},{"x0":1,"y0":1},{"x0":2,"y0":4},{"x0":3,"y0":9},{"x0":4,"y0":16},{"x0":5,"y0":25}]}],"scales":[{"name":"scatter_x","type":"linear","domain":{"data":"scatter","field":"data.x0"},"range":"width"},{"name":"scatter_y","type":"linear","domain":{"data":"scatter","field":"data.y0"},"range":"height"}],"marks":[{"type":"symbol","from":{"data":"scatter"},"properties":{"enter":{"size":{"value":20},"shape":{"value":"circle"},"x":{"field":"data.x0","scale":"scatter_x"},"y":{"field":"data.y0","scale":"scatter_y"},"fill":{"value":"blue"}}}}],"axes":[{"type":"x","scale":"scatter_x"},{"type":"y","scale":"scatter_y"}]}'
+    end
+
+    it "should correctly make a scatter plot with default options and multiple x and y data series" do
+      @xdata = [[0,1,2,3,4,16], [10,11,12,13,14,15]]
+      Plotrb::Simple.scatter(@xdata, @ydata).generate_spec.should eq '{"width":640,"height":480,"data":[{"name":"scatter","values":[{"x0":0,"y0":0,"x1":10,"y1":0},{"x0":1,"y0":1,"x1":11,"y1":1},{"x0":2,"y0":4,"x1":12,"y1":8},{"x0":3,"y0":9,"x1":13,"y1":27},{"x0":4,"y0":16,"x1":14,"y1":64},{"x0":16,"y0":25,"x1":15,"y1":125}]}],"scales":[{"name":"scatter_x","type":"linear","domain":{"data":"scatter","field":"data.x0"},"range":"width"},{"name":"scatter_y","type":"linear","domain":{"data":"scatter","field":"data.y0"},"range":"height"}],"marks":[{"type":"symbol","from":{"data":"scatter"},"properties":{"enter":{"size":{"value":20},"shape":{"value":"circle"},"x":{"field":"data.x0","scale":"scatter_x"},"y":{"field":"data.y0","scale":"scatter_y"},"fill":{"value":"blue"}}}},{"type":"symbol","from":{"data":"scatter"},"properties":{"enter":{"size":{"value":20},"shape":{"value":"circle"},"x":{"field":"data.x1","scale":"scatter_x"},"y":{"field":"data.y1","scale":"scatter_y"},"fill":{"value":"blue"}}}}],"axes":[{"type":"x","scale":"scatter_x"},{"type":"y","scale":"scatter_y"}]}'
+    end
+  end
+end
+
+

data/spec/plotrb/transforms_spec.rb

@@ -0,0 +1,248 @@
+require_relative '../spec_helper'
+
+describe 'Transform' do
+
+  describe '#initialize' do
+
+    subject { Plotrb::Transform }
+
+    it 'calls respective type method to initialize attributes' do
+      subject.any_instance.should_receive(:send).with(:foo)
+      subject.new(:foo)
+    end
+
+    it 'raises error if type is not recognized' do
+      expect { subject.new(:foo) }.to raise_error(NoMethodError)
+    end
+
+  end
+
+  describe '#array' do
+
+    subject { Plotrb::Transform.new(:array) }
+
+    it 'responds to #take' do
+      subject.take('foo', 'bar')
+      subject.send(:process_array_fields)
+      subject.fields.should match_array(['data.foo', 'data.bar'])
+    end
+
+  end
+
+  describe '#copy' do
+
+    subject { Plotrb::Transform.new(:copy) }
+
+    it 'responds to #take' do
+      subject.take('foo_field', 'bar_field').from('some_data').as('foo', 'bar')
+      subject.fields.should match_array(['foo_field', 'bar_field'])
+    end
+
+    it 'raises error if as and fields are of different size' do
+      subject.take('foo', 'bar').from('data').as('baz')
+      expect { subject.send(:process_copy_as) }.to raise_error(ArgumentError)
+    end
+
+  end
+
+  describe '#cross' do
+
+    subject { Plotrb::Transform.new(:cross) }
+
+    it 'raises error if the secondary data does not exist' do
+      subject.with('foo')
+      ::Plotrb::Kernel.stub(:find_data).and_return(nil)
+      expect { subject.send(:process_cross_with) }.to raise_error(ArgumentError)
+    end
+
+  end
+
+  describe '#facet' do
+
+    subject { Plotrb::Transform.new(:facet) }
+
+    it 'responds to #group_by' do
+      subject.group_by('foo', 'bar')
+      subject.send(:process_facet_keys)
+      subject.keys.should match_array(['data.foo', 'data.bar'])
+    end
+
+  end
+
+  describe '#filter' do
+
+    subject { Plotrb::Transform.new(:filter) }
+
+    it 'adds variable d if not present in the test expression' do
+      subject.test('x>10')
+      expect { subject.send(:process_filter_test) }.to raise_error(ArgumentError)
+    end
+
+  end
+
+  describe '#flatten' do
+
+    subject { Plotrb::Transform.new(:flatten) }
+
+  end
+
+  describe '#fold' do
+
+    subject { Plotrb::Transform.new(:fold) }
+
+    it 'responds to #into' do
+      subject.into('foo', 'bar')
+      subject.send(:process_fold_fields)
+      subject.fields.should match_array(['data.foo', 'data.bar'])
+    end
+
+  end
+
+  describe '#formula' do
+
+    subject { Plotrb::Transform.new(:formula) }
+
+    it 'responds to #apply and #into' do
+      subject.apply('some_expression').into('some_field')
+      subject.field.should == 'some_field'
+      subject.expr.should == 'some_expression'
+    end
+
+  end
+
+  describe '#slice' do
+
+    subject { Plotrb::Transform.new(:slice) }
+
+    it 'slices by a single value'
+
+    it 'slices by a range'
+
+    it 'slices by special positions'
+
+    it 'raises error otherwise'
+
+  end
+
+  describe '#sort' do
+
+    subject { Plotrb::Transform.new(:sort) }
+
+    it 'adds - in front for reverse sort'
+
+  end
+
+  describe '#stats' do
+
+    subject { Plotrb::Transform.new(:stats) }
+
+    it 'responds to #from, #include_median, and #store_stats' do
+      subject.from('foo').include_median.store_stats
+      subject.value.should == 'foo'
+      subject.median.should be_true
+      subject.assign.should be_true
+    end
+
+  end
+
+  describe '#truncate' do
+
+    subject { Plotrb::Transform.new(:truncate) }
+
+    it 'responds to #from, #to, and #max_length' do
+      subject.from('foo').to('bar').max_length(5)
+      subject.send(:process_truncate_value)
+      subject.value.should == 'data.foo'
+      subject.output.should == 'bar'
+      subject.limit.should == 5
+    end
+
+    it 'responds to #in_position' do
+      subject.in_front
+      subject.position.should == :front
+    end
+
+  end
+
+  describe '#unique' do
+
+    subject { Plotrb::Transform.new(:unique) }
+
+    it 'responds to #from and #to' do
+      subject.from('foo').to('bar')
+      subject.send(:process_unique_field)
+      subject.field.should == 'data.foo'
+      subject.as.should == 'bar'
+    end
+
+  end
+
+  describe '#window' do
+
+    subject { Plotrb::Transform.new(:window) }
+
+  end
+
+  describe '#zip' do
+
+    subject { Plotrb::Transform.new(:zip) }
+
+    it 'responds to #match and #against' do
+      subject.with('foo').as('bar').match('foo_field').against('bar_field')
+      subject.send(:process_zip_key)
+      subject.send(:process_zip_with_key)
+      subject.key.should == 'data.foo_field'
+      subject.with_key.should == 'data.bar_field'
+    end
+
+  end
+
+  describe '#force' do
+
+    subject { Plotrb::Transform.new(:force) }
+
+  end
+
+  describe '#geo' do
+
+    subject { Plotrb::Transform.new(:geo) }
+
+  end
+
+  describe '#geopath' do
+
+    subject { Plotrb::Transform.new(:geopath) }
+
+  end
+
+  describe '#link' do
+
+    subject { Plotrb::Transform.new(:link) }
+
+  end
+
+  describe '#pie' do
+
+    subject { Plotrb::Transform.new(:pie) }
+
+  end
+
+  describe '#stack' do
+
+    subject { Plotrb::Transform.new(:stack) }
+
+  end
+
+  describe '#treemap' do
+
+    subject { Plotrb::Transform.new(:treemap) }
+
+  end
+
+  describe '#wordcloud' do
+
+    subject { Plotrb::Transform.new(:wordcloud) }
+
+  end
+
+end