bm25f 0.1.0 → 0.2.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: bc3b5ddbf5d62479a1a0afafccef1d377db2a149ac9db87f8d207b0c16b41550
-  data.tar.gz: 295be306f71a84ae399cffaf7c1faff3a768ad7e1c00095230144d564cc9af59
+  metadata.gz: 2649ac16a2f5adeb5815b2f472e62f98aec6b05af882648e4d128051c5fb90e0
+  data.tar.gz: e70ee90881438943874b4ae5e4852a6e7d44eda53f0315af35ea85565a2cde70
 SHA512:
-  metadata.gz: 1c06abf6c6d53a66e151378c610bf5ecfbe92ced01692bbe126d7bfda77bd85a2c2733934460052a6d8129e61881e067a1e35264f7acb197ff51f2336bb31e19
-  data.tar.gz: 85ad4ba68f49009bbe667ccf993d498671ced59f9aeff4394435778c53369037f7dee346b7111c3cc6679fc72b9c8e62798139d3dce7a56e098795844e59b85e
+  metadata.gz: 1f7a68810a77daa256eb3e3177748e6b22f53cf3dfd2e68c68e82c4b7c143cdc60ac38c0e519e6f4b3c90725123c62f2b5e852898230c5e9812ca181e3efa577
+  data.tar.gz: 6690099e9435eebc6676d7bbd0d5530218e2d360a2ca8fb3e5156faf74b426364765eab60bdab5bb2caa85deadb69bb22b951a987d9904eaf1cb75bbb1b31520

data/test/test_bm25f.rb

@@ -0,0 +1,23 @@
+require 'minitest/autorun'
+require 'bm25f'
+
+class BM25FTest < Minitest::Test
+  def setup
+    @bm25f = BM25F.new
+    @documents = [
+      { title: 'hello world', content: 'foo bar baz' },
+      { title: 'foo bar', content: 'goodbye, world!' }
+    ]
+  end
+
+  def test_score
+    @bm25f.fit @documents
+    scores = @bm25f.score 'hello world foo bar baz'
+
+    # Sort
+    scores = scores.to_a.sort_by { |_, v| v.to_i }
+
+    # Checks if the most matching element is the first element
+    assert scores.last[0].zero?
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: bm25f
 version: !ruby/object:Gem::Version
-  version: 0.1.0
+  version: 0.2.0
 platform: ruby
 authors:
 - catflip
@@ -24,6 +24,20 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: '2.1'
+- !ruby/object:Gem::Dependency
+  name: rake
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
+  type: :development
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: '0'
 description: A fast implementation of the BM25F ranking algorithm for information
   retrieval systems, written in Ruby.
 email:
@@ -32,6 +46,7 @@ extensions: []
 extra_rdoc_files: []
 files:
 - lib/bm25f.rb
+- test/test_bm25f.rb
 homepage: https://github.com/catflip/bm25f-ruby
 licenses:
 - AGPL-3.0