pest 0.0.0

data/spec/pest/function/probability_spec.rb

@@ -0,0 +1,118 @@
+require 'spec_helper'
+
+class ProbabilityTestClass
+  include Pest::Estimator
+  include Pest::Function::Probability
+end
+
+describe Pest::Function::Probability do
+  before(:each) do
+    @v1 = Pest::Variable.new(:name => :foo)
+    @v2 = Pest::Variable.new(:name => :bar)
+    @v3 = Pest::Variable.new(:name => :baz)
+    @instance = ProbabilityTestClass.new
+    @instance.stub(:variables).and_return({:foo => @v1, :bar => @v2})
+  end
+
+  describe "#probability" do
+    it "returns a Builder" do
+      @instance.probability.should be_a(Pest::Function::Probability::Builder)
+    end
+
+    it "is aliased as p" do
+      @instance.p.should be_a(Pest::Function::Probability::Builder)
+    end
+  end
+
+  describe Pest::Function::Probability::Builder do
+    describe "::new" do
+      before(:each) { @builder = ProbabilityTestClass::Builder.new(@instance, [@v1, :bar]) }
+
+      it "sets estimator" do
+        @builder.estimator.should == @instance
+      end
+
+      it "sets event" do
+        @builder.event.should == [@v1, @v2].to_set
+      end
+
+      it "fails if variable undefined for estimator" do
+        lambda { ProbabilityTestClass::Builder.new(@instance, [@v1, @v3]) }.should raise_error(ArgumentError)
+      end
+
+      it "constructs dataset if passed hash"
+    end
+
+    describe "#given" do
+      before(:each) { @builder = ProbabilityTestClass::Builder.new(@instance, [:foo]) }
+
+      it "sets givens" do
+        @builder.given(:bar)
+        @builder.givens.should include(@v2)
+      end
+
+      it "returns self" do
+        @builder.given(:bar).should be_a(ProbabilityTestClass::Builder)
+      end
+
+      it "fails if variables aren't variables on the estimator" do
+        lambda { @builder.given(:baz) }.should raise_error(ArgumentError)
+      end
+
+      it "adds to dataset if passed hash"
+
+      it "raises error if passed hash with existing (non hash) dataset"
+    end
+
+    describe "#in" do
+      it "sets data source" do
+        data_set = double('DataSet')
+        ProbabilityTestClass::Builder.new(@instance,[:foo]).in(data_set).data_source.should == data_set
+      end
+
+      it "raises error if existing data source"
+    end
+
+    describe "#evaluate" do
+      it "generates dataset if not specified"
+
+      it "gets probability of event" do
+        event = double('EventDist')
+        @instance.distributions.stub(:[]).with([@v1].to_set).and_return(event)
+        event.should_receive(:probability).and_return 0.5
+
+        ProbabilityTestClass::Builder.new(@instance,[:foo]).evaluate
+      end
+
+      it "gets probability of givens" do
+        event = double('EventDist')
+        given = double('GivenDist')
+        @instance.distributions.stub(:[]).with([@v1].to_set).and_return(event)
+        @instance.distributions.stub(:[]).with([@v2].to_set).and_return(given)
+        event.stub(:probability).and_return 0.5
+        given.should_receive(:probability).and_return 0.5
+
+        ProbabilityTestClass::Builder.new(@instance,[:foo]).given(:bar).evaluate
+      end
+
+      it "returns Pr event / givens (if givens)" do
+        event = double('EventDist')
+        given = double('GivenDist')
+        @instance.distributions.stub(:[]).with([@v1].to_set).and_return(event)
+        @instance.distributions.stub(:[]).with([@v2].to_set).and_return(given)
+        event.stub(:probability).and_return 0.5
+        given.stub(:probability).and_return 0.5
+
+        ProbabilityTestClass::Builder.new(@instance,[:foo]).given(:bar).evaluate.should == 1.0
+      end
+
+      it "returns Pr event (if no givens)" do
+        event = double('EventDist')
+        @instance.distributions.stub(:[]).with([@v1].to_set).and_return(event)
+        event.stub(:probability).and_return 0.5
+
+        ProbabilityTestClass::Builder.new(@instance,[:foo]).evaluate.should == 0.5
+      end
+    end
+  end
+end

data/spec/pest/variable_spec.rb

@@ -0,0 +1,73 @@
+require 'spec_helper'
+
+describe Pest::Variable do
+  before(:each) { @class = Pest::Variable }
+
+  describe "#uuid" do
+    it "is generated if not specified" do
+      @class.new.uuid.should be_true
+    end
+
+    it "uses uuid passed at instantiation" do
+      @class.new(:uuid => 'foo').uuid.should == 'foo'
+    end
+  end
+
+  describe "#name" do
+    it "has it" do
+      @class.new(:name => 'foo').name.should == 'foo'
+    end
+  end
+
+  describe "#identifier" do
+    before(:each) do
+      @instance = @class.new(:name => 'foo', :uuid => 'bar')
+    end
+
+    it "concatenates name:uuid" do
+      @instance.identifier.should == 'foo:bar'
+    end
+
+    it "is aliased as serialize" do
+      @instance.serialize.should == 'foo:bar'
+    end
+  end
+
+  describe "#==" do
+    before(:each) do
+      @one = @class.new(:name => 'foo', :uuid => 'bar')
+      @two = @class.new(:name => 'foo', :uuid => 'bar')
+      @three = @class.new(:name => 'foo', :uuid => 'baz')
+    end
+
+    it "returns true if same identifier" do
+      @one.should == @two
+    end
+
+    it "returns false if identifier doesn't match" do
+      @one.should_not == @three
+    end
+
+    it "returns false if not a Variable instance" do
+      @one.should_not == 'hello'
+    end
+
+    it "works for sets" do
+      [@one].to_set.should == [@two].to_set
+    end
+  end
+
+  describe "::deserialize" do
+    before(:each) do
+      @instance = @class.deserialize('foo:4f9b5243-2e2e-4a90-8009-21079fad5855')
+    end
+
+    it "instantiates with name" do
+      @instance.name.should == 'foo'
+    end
+
+    it "instantiates with uuid" do
+      @instance.uuid.should == '4f9b5243-2e2e-4a90-8009-21079fad5855'
+    end
+  end
+end

data/spec/pest_spec.rb

@@ -0,0 +1,4 @@
+require 'spec_helper'
+
+describe Pest do
+end

data/spec/spec_helper.rb

@@ -0,0 +1,14 @@
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'rspec'
+require 'pest'
+require 'pry'
+
+# Requires supporting files with custom matchers and macros, etc,
+# in ./support/ and its subdirectories.
+Dir["#{File.dirname(__FILE__)}/support/**/*.rb"].each {|f| require f}
+
+RSpec.configure do |config|
+  config.color_enabled = true
+  config.formatter = :documentation
+end

metadata

@@ -0,0 +1,321 @@
+--- !ruby/object:Gem::Specification
+name: pest
+version: !ruby/object:Gem::Version
+  version: 0.0.0
+  prerelease: 
+platform: ruby
+authors:
+- Ryan Michael
+autorequire: 
+bindir: bin
+cert_chain: []
+date: 2012-06-24 00:00:00.000000000 Z
+dependencies:
+- !ruby/object:Gem::Dependency
+  name: pest
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: narray
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: uuidtools
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: jeweler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: jeweler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: jeweler
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: pry
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    none: false
+    requirements:
+    - - ! '>='
+      - !ruby/object:Gem::Version
+        version: '0'
+description: Wrappers to facilitate different classes of probability estimators
+email: kerinin@gmail.com
+executables: []
+extensions: []
+extra_rdoc_files:
+- README.md
+files:
+- .travis.yml
+- Gemfile
+- README.md
+- Rakefile
+- VERSION
+- lib/pest.rb
+- lib/pest/data_set.rb
+- lib/pest/data_set/hash.rb
+- lib/pest/data_set/narray.rb
+- lib/pest/estimator.rb
+- lib/pest/estimator/frequency.rb
+- lib/pest/function.rb
+- lib/pest/function/entropy.rb
+- lib/pest/function/probability.rb
+- lib/pest/variable.rb
+- lib/pest/version.rb
+- pest.gemspec
+- spec/pest/data_set/hash_spec.rb
+- spec/pest/data_set/narray_spec.rb
+- spec/pest/data_set_spec.rb
+- spec/pest/estimator/bernoulli_spec.rb
+- spec/pest/estimator/frequency_spec.rb
+- spec/pest/estimator/gaussian_spec.rb
+- spec/pest/estimator/multinomial_spec.rb
+- spec/pest/estimator/parzen_spec.rb
+- spec/pest/estimator/svd_spec.rb
+- spec/pest/estimator_spec.rb
+- spec/pest/function/entropy_spec.rb
+- spec/pest/function/probability_spec.rb
+- spec/pest/variable_spec.rb
+- spec/pest_spec.rb
+- spec/spec_helper.rb
+homepage: http://github.com/kerinin/pest
+licenses:
+- MIT
+post_install_message: 
+rdoc_options: []
+require_paths:
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+      segments:
+      - 0
+      hash: 1228436893259632292
+required_rubygems_version: !ruby/object:Gem::Requirement
+  none: false
+  requirements:
+  - - ! '>='
+    - !ruby/object:Gem::Version
+      version: '0'
+requirements: []
+rubyforge_project: 
+rubygems_version: 1.8.24
+signing_key: 
+specification_version: 3
+summary: Probability Estimation
+test_files: []