inferx 0.2.2 → 0.2.3

data/lib/inferx/adapter.rb

@@ -55,7 +55,7 @@ class Inferx
 
     protected
 
-    %w(hdel hget hgetall hincrby hkeys hsetnx).each do |command|
+    %w(hdel hexists hget hgetall hincrby hkeys hsetnx).each do |command|
       define_method(command) do |*args|
         @redis.__send__(command, categories_key, *args)
       end

data/lib/inferx/categories.rb

@@ -44,6 +44,14 @@ class Inferx
       end
     end
 
+    # Determine if the category is defined.
+    #
+    # @param [String] category_name a category name
+    # @return whether the category is defined
+    def exists?(category_name)
+      hexists(category_name)
+    end
+
     # Apply process for each category.
     #
     # @yield a block to be called for every category

data/lib/inferx/category.rb

@@ -3,6 +3,14 @@ require 'inferx/adapter'
 class Inferx
   class Category < Adapter
 
+    def self.ready_for(method_name)
+      define_method("ready_to_#{method_name}") do |&block|
+        all = []
+        block[lambda { |items| all += items }]
+        __send__(method_name, all)
+      end
+    end
+
     # @param [Redis] redis an instance of Redis
     # @param [String] name a category name
     # @param [Integer] size total of scores
@@ -75,9 +83,7 @@ class Inferx
     #
     # @yield [train] process something
     # @yieldparam [Proc] train enhance the training data giving words
-    def ready_to_train(&process)
-      train(aggregate(&process))
-    end
+    ready_for :train
 
     # Attenuate the training data giving words.
     #
@@ -114,9 +120,7 @@ class Inferx
     #
     # @yield [untrain] process something
     # @yieldparam [Proc] untrain attenuate the training data giving words
-    def ready_to_untrain(&process)
-      untrain(aggregate(&process))
-    end
+    ready_for :untrain
 
     # Get effectively scores for each word.
     #
@@ -143,11 +147,5 @@ class Inferx
         hash
       end
     end
-
-    def aggregate
-      all = []
-      yield lambda { |items| all += items }
-      all
-    end
   end
 end

data/lib/inferx/complementary/category.rb

@@ -4,6 +4,32 @@ class Inferx
   module Complementary
     class Category < Inferx::Category
 
+      # Inject the words to the training data of the category.
+      #
+      # @param [Array<String>] words an array of words
+      alias inject train
+
+      # Eject the words from the training data of the category.
+      #
+      # @param [Array<String>] words an array of words
+      alias eject untrain
+
+      # Prepare to inject the words to the training data of the category. Use
+      # for high performance.
+      #
+      # @yield [train] process something
+      # @yieldparam [Proc] inject inject the words to the training data of the
+      #   category
+      ready_for :inject
+
+      # Prepare to eject the words from the training data of the category. Use
+      # for high performance.
+      #
+      # @yield [train] process something
+      # @yieldparam [Proc] eject eject the words from the training data of the
+      #   category
+      ready_for :eject
+
       # Enhance the training data of other categories giving words.
       #
       # @param [Array<String>] words an array of words

data/lib/inferx/version.rb

@@ -1,3 +1,3 @@
 class Inferx
-  VERSION = '0.2.2'
+  VERSION = '0.2.3'
 end

data/spec/inferx/categories_spec.rb

@@ -151,6 +151,35 @@ describe Inferx::Categories, '#remove' do
   end
 end
 
+describe Inferx::Categories, '#exists?' do
+  it 'calls Redis#hexists' do
+    redis = redis_stub.tap do |s|
+      s.should_receive(:hexists).with('inferx:categories', 'red')
+    end
+
+    categories = described_class.new(redis)
+    categories.exists?('red')
+  end
+
+  it 'returns true if the category is defined' do
+    redis = redis_stub.tap do |s|
+      s.stub!(:hexists => true)
+    end
+
+    categories = described_class.new(redis)
+    categories.should be_exists('red')
+  end
+
+  it 'returns false if the category is not defined' do
+    redis = redis_stub.tap do |s|
+      s.stub!(:hexists => false)
+    end
+
+    categories = described_class.new(redis)
+    categories.should_not be_exists('red')
+  end
+end
+
 describe Inferx::Categories, '#each' do
   before do
     @redis = redis_stub do |s|

data/spec/inferx/complementary/category_spec.rb

@@ -1,5 +1,31 @@
 require 'inferx/complementary/category'
 
+describe Inferx::Complementary::Category, '#ready_to_inject' do
+  it 'calls #inject with the words to inject block' do
+    category = described_class.new(redis_stub, 'red', 2)
+    category.should_receive(:inject).with(%w(word1 word2 word3))
+
+    category.ready_to_inject do |inject|
+      inject[%w(word1)]
+      inject[%w(word2)]
+      inject[%w(word3)]
+    end
+  end
+end
+
+describe Inferx::Complementary::Category, '#ready_to_eject' do
+  it 'calls #eject with the words to eject block' do
+    category = described_class.new(redis_stub, 'red', 2)
+    category.should_receive(:eject).with(%w(word1 word2 word3))
+
+    category.ready_to_eject do |eject|
+      eject[%w(word1)]
+      eject[%w(word2)]
+      eject[%w(word3)]
+    end
+  end
+end
+
 describe Inferx::Complementary::Category, '#train' do
   it 'calls Redis#zincrby and Redis#hincrby for other categories' do
     redis = redis_stub do |s|

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: inferx
 version: !ruby/object:Gem::Version
-  version: 0.2.2
+  version: 0.2.3
   prerelease: 
 platform: ruby
 authors:
@@ -9,7 +9,7 @@ authors:
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2012-10-22 00:00:00.000000000 Z
+date: 2012-10-23 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: redis