Merge pull request #38 from cardmagic/add-bayes-tests

Add test for classifying safari animals

Author: cardmagic

.github/workflows/ruby.yml

@@ -1,34 +1,38 @@
+# This workflow uses actions that are not certified by GitHub.
+# They are provided by a third-party and are governed by
+# separate terms of service, privacy policy, and support
+# documentation.
+# This workflow will download a prebuilt Ruby version, install dependencies and run tests with Rake
+# For more information see: https://github.com/marketplace/actions/setup-ruby-jruby-and-truffleruby
+
 name: Ruby
 
 on:
   push:
-    branches:
-      - main
+    branches: [ "master" ]
   pull_request:
-    branches:
-      - main
+    branches: [ "master" ]
+
+permissions:
+  contents: read
 
 jobs:
   test:
+
     runs-on: ubuntu-latest
     strategy:
       matrix:
-        ruby-version: [2.7, 3.1, 3.2, 3.3] # Add the Ruby versions to test against
+        ruby-version: ['2.7', '3.0', '3.1', '3.2', 'head']
 
     steps:
-      - name: Checkout code
-        uses: actions/checkout@v3
-
-      - name: Set up Ruby
-        uses: ruby/setup-ruby@v1
-        with:
-          ruby-version: ${{ matrix.ruby-version }}
-
-      - name: Install dependencies
-        run: |
-          gem install bundler
-          bundle install
-
-      - name: Run tests
-        run: |
-          bundle exec rake test
+    - uses: actions/checkout@v4
+    - name: Set up Ruby
+    # To automatically get bug fixes and new Ruby versions for ruby/setup-ruby,
+    # change this to (see https://github.com/ruby/setup-ruby#versioning):
+    # uses: ruby/setup-ruby@v1
+      uses: ruby/setup-ruby@55283cc23133118229fd3f97f9336ee23a179fcf # v1.146.0
+      with:
+        ruby-version: ${{ matrix.ruby-version }}
+        bundler-cache: true # runs 'bundle install' and caches installed gems automatically
+    - name: Run tests
+      run: bundle exec rake test

test/bayes/bayesian_test.rb

@@ -27,4 +27,19 @@ def test_classification
     @classifier.train_uninteresting 'here are some bad words, I hate you'
     assert_equal 'Uninteresting', @classifier.classify('I hate bad words and you')
   end
+
+  def test_safari_animals
+    bayes = Classifier::Bayes.new 'Lion', 'Elephant'
+    bayes.train_lion 'lion'
+    bayes.train_lion 'zebra'
+    bayes.train_elephant 'elephant'
+    bayes.train_elephant 'trunk'
+    bayes.train_elephant 'tusk'
+
+    assert_equal 'Lion', bayes.classify('zebra')
+    assert_equal 'Elephant', bayes.classify('trunk')
+    assert_equal 'Elephant', bayes.classify('tusk')
+    assert_equal 'Lion', bayes.classify('lion')
+    assert_equal 'Elephant', bayes.classify('elephant')
+  end
 end