bayesball 0.1.1 → 0.2.0

checksums.yaml

@@ -0,0 +1,7 @@
+---
+SHA1:
+  metadata.gz: 478528e5894a3dedc57435782136dae425e36c49
+  data.tar.gz: 2c6b799680658bb607031eee0b4ca985eff6f14a
+SHA512:
+  metadata.gz: f28d402de88b753a1fe012e5e7e3cabdf59a8c001d6154fcd02d237fe1d66462922e50dffd9f18cb6900d92d0be4d6dd5a98c148d5c0686a4e6d11651fc7c2cb
+  data.tar.gz: 804e9ebdc8a73a5fe741482ffa2282624227530b136fac9a1faf52bee0b0f0e4dde152e163a104b9d47f267a44171d97d4d25a63200606aea1d31899cd519bf7

data/Gemfile

@@ -3,8 +3,9 @@ source 'https://rubygems.org'
 gemspec
 
 group :development do
-  gem 'guard-minitest'
   gem 'rake'
   gem 'mongo'
   gem 'bson_ext'
+  gem 'sequel'
+  gem 'sqlite3'
 end

data/lib/bayesball/persistence.rb

@@ -1,5 +1,6 @@
 module Bayesball
   module Persistence
     autoload :Mongo, 'bayesball/persistence/mongo'
+    autoload :Sequel, 'bayesball/persistence/sequel'
   end
 end

data/lib/bayesball/persistence/sequel.rb

@@ -0,0 +1,38 @@
+module Bayesball
+  module Persistence
+    class Sequel
+      include Enumerable
+
+      attr_reader :dataset
+      def initialize(dataset)
+        @dataset = dataset
+      end
+
+      def empty?
+        @collection.count == 0
+      end
+
+      def [](key)
+        dataset.where(bucket: key).to_hash(:word, :count)
+      end
+
+      def []=(key, value)
+        dataset.where(bucket: key).delete
+        value.each do |word, count|
+          dataset.insert word: word, count: count, bucket: key
+        end
+      end
+
+      def each
+        if block_given?
+          dataset.to_a.group_by{|i| i[:bucket] }.each do |bucket, group|
+            counts = Hash[group.map{|g| [g[:word], g[:count]]}]
+            yield [bucket, counts]
+          end
+        else
+          self.to_enum
+        end
+      end
+    end
+  end
+end

data/lib/bayesball/version.rb

@@ -1,3 +1,3 @@
 module Bayesball
-  VERSION = "0.1.1"
+  VERSION = "0.2.0"
 end

data/spec/bayesball/persistence/sequel_spec.rb

@@ -0,0 +1,38 @@
+require 'spec_helper'
+
+module Bayesball
+  module Persistence
+    describe Sequel do
+      let(:dataset) { SEQUEL_DB[:bayesball_buckets] }
+      let(:persistence) { Sequel.new(dataset) }
+
+      it 'should return counts' do
+        persistence['truck'].wont_be_nil
+      end
+
+      it 'should set counts' do
+        persistence['car'] = {'x' => 3, 'y' => 2}
+        persistence['car'].must_equal 'x' => 3, 'y' => 2
+      end
+
+      it 'should reduce' do
+        persistence['duck'] = {'x' => 8, 'y' => 5}
+        persistence['cat'] = {'x' => 7, 'y' => 1}
+        persistence['dog'] = {'x' => 2, 'y' => 2}
+
+        result = persistence.reduce({}) do |memo, (category, counts)|
+          total = counts.values.reduce(:+).to_f
+          memo[category] = total
+          memo
+        end
+        result['duck'].must_equal 13
+        result['cat'].must_equal 8
+        result['dog'].must_equal 4
+      end
+
+      it 'should return enumerator from each' do
+         persistence.each.class.must_equal Enumerator
+      end
+    end
+  end
+end

data/spec/spec_helper.rb

@@ -7,3 +7,14 @@ require 'minitest/pride'
 require 'minitest/autorun'
 
 Bundler.require :development
+
+SEQUEL_DB = Sequel.connect 'sqlite:///'
+
+SEQUEL_DB.create_table :bayesball_buckets do
+  column :bucket, String, null: false
+  column :word, String, null: false
+  column :count, Integer, null: false
+
+  index :bucket
+  index [:bucket, :word], unique: true
+end

metadata

@@ -1,15 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: bayesball
 version: !ruby/object:Gem::Version
-  version: 0.1.1
-  prerelease: 
+  version: 0.2.0
 platform: ruby
 authors:
 - Jason Staten
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-04-13 00:00:00.000000000 Z
+date: 2013-08-13 00:00:00.000000000 Z
 dependencies: []
 description: A bayes classifier
 email:
@@ -29,42 +28,38 @@ files:
 - lib/bayesball/classifier.rb
 - lib/bayesball/persistence.rb
 - lib/bayesball/persistence/mongo.rb
+- lib/bayesball/persistence/sequel.rb
 - lib/bayesball/stopwords.txt
 - lib/bayesball/version.rb
 - spec/bayesball/classifier_spec.rb
 - spec/bayesball/persistence/mongo_spec.rb
+- spec/bayesball/persistence/sequel_spec.rb
 - spec/spec_helper.rb
 homepage: ''
 licenses: []
+metadata: {}
 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: -4130593485830818059
 required_rubygems_version: !ruby/object:Gem::Requirement
-  none: false
   requirements:
-  - - ! '>='
+  - - '>='
     - !ruby/object:Gem::Version
       version: '0'
-      segments:
-      - 0
-      hash: -4130593485830818059
 requirements: []
 rubyforge_project: 
-rubygems_version: 1.8.18
+rubygems_version: 2.0.3
 signing_key: 
-specification_version: 3
+specification_version: 4
 summary: A bayes classifier
 test_files:
 - spec/bayesball/classifier_spec.rb
 - spec/bayesball/persistence/mongo_spec.rb
+- spec/bayesball/persistence/sequel_spec.rb
 - spec/spec_helper.rb