scout-ai 1.1.6 → 1.1.7

data/test/scout/network/test_knowledge_base.rb

@@ -0,0 +1,61 @@
+require File.expand_path(__FILE__).sub(%r(/test/.*), '/test/test_helper.rb')
+require File.expand_path(__FILE__).sub(%r(.*/test/), '').sub(/test_(.*)\.rb/,'\\1')
+
+require 'scout/network/knowledge_base'
+
+class TestKnowledgeBaseRadialExpand < Test::Unit::TestCase
+  def test_radial_expand_outward
+    TmpFile.with_dir do |dir|
+      kb = KnowledgeBase.new dir
+      kb.register :brothers, datafile_test(:person).brothers, undirected: true
+
+      seeds = ["Miki"]
+      visited, edges = kb.radial_expand(:brothers, seeds, depth: 1, direction: :out)
+
+      assert_include visited, "Miki"
+      assert_include visited, "Isa"
+      assert edges.any? { |e| [e.source, e.target].sort == %w[Isa Miki].sort }
+    end
+  end
+
+  def test_radial_layers
+    TmpFile.with_dir do |dir|
+      kb = KnowledgeBase.new dir
+      kb.register :brothers, datafile_test(:person).brothers, undirected: true
+
+      seeds = ["Miki"]
+      layers, _edges = kb.radial_layers(:brothers, seeds, depth: 2, direction: :out)
+
+      # layers[0] should be seeds
+      assert_equal [seeds], layers.first(1)
+      # At least Isa should appear in layer 1
+      assert_include layers[1], "Isa"
+    end
+  end
+
+  def test_radial_expand_filter
+    TmpFile.with_dir do |dir|
+      kb = KnowledgeBase.new dir
+      kb.register :brothers, datafile_test(:person).brothers, undirected: true
+
+      seeds = ["Miki"]
+      visited, edges = kb.radial_expand(:brothers, seeds, depth: 1, direction: :out,
+                                        filter: ->(item){ false })
+
+      assert_equal ["Miki"], visited
+      assert_equal [], edges
+    end
+  end
+
+  def test_subgraph_around
+    TmpFile.with_dir do |dir|
+      kb = KnowledgeBase.new dir
+      kb.register :brothers, datafile_test(:person).brothers, undirected: true
+
+      seeds = ["Miki"]
+      edges = kb.subgraph_around(:brothers, seeds, depth: 1, direction: :out)
+
+      assert edges.any? { |e| [e.source, e.target].sort == %w[Isa Miki].sort }
+    end
+  end
+end

data/test/scout/network/test_paths.rb

@@ -0,0 +1,56 @@
+require File.expand_path(__FILE__).sub(%r(/test/.*), '/test/test_helper.rb')
+require File.expand_path(__FILE__).sub(%r(.*/test/), '').sub(/test_(.*)\.rb/,'\\1')
+
+require 'scout/network/paths'
+
+class TestNetworkPaths < Test::Unit::TestCase
+  def test_breadth_first_distances
+    adjacency = {
+      'A' => %w[B C],
+      'B' => %w[D],
+      'C' => %w[D E],
+      'D' => %w[F],
+      'E' => [],
+      'F' => []
+    }
+
+    distances = Paths.breadth_first(adjacency, 'A', 2)
+
+    assert_equal 0, distances['A']
+    assert_equal 1, distances['B']
+    assert_equal 1, distances['C']
+    assert_equal 2, distances['D']
+    assert_equal 2, distances['E']
+    # F should not be reached within 2 steps
+    assert_nil distances['F']
+  end
+
+  def test_neighborhood_from_single_source
+    adjacency = {
+      'A' => %w[B C],
+      'B' => %w[D],
+      'C' => %w[D E],
+      'D' => %w[F],
+      'E' => [],
+      'F' => []
+    }
+
+    neigh = Paths.neighborhood(adjacency, 'A', 2)
+
+    # Order is not guaranteed; compare as sets
+    assert_equal %w[A B C D E].sort, neigh.sort
+  end
+
+  def test_neighborhood_from_multiple_sources
+    adjacency = {
+      'A' => %w[B],
+      'B' => %w[C],
+      'X' => %w[Y],
+      'Y' => %w[Z]
+    }
+
+    neigh = Paths.neighborhood(adjacency, %w[A X], 1)
+
+    assert_equal %w[A B X Y].sort, neigh.sort
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: scout-ai
 version: !ruby/object:Gem::Version
-  version: 1.1.6
+  version: 1.1.7
 platform: ruby
 authors:
 - Miguel Vazquez
@@ -147,6 +147,9 @@ files:
 - lib/scout/model/python/torch/load_and_save.rb
 - lib/scout/model/util/run.rb
 - lib/scout/model/util/save.rb
+- lib/scout/network/entity.rb
+- lib/scout/network/knowledge_base.rb
+- lib/scout/network/paths.rb
 - python/scout_ai/__init__.py
 - python/scout_ai/huggingface/data.py
 - python/scout_ai/huggingface/eval.py
@@ -203,6 +206,9 @@ files:
 - test/scout/model/python/torch/test_helpers.rb
 - test/scout/model/test_base.rb
 - test/scout/model/util/test_save.rb
+- test/scout/network/test_entity.rb
+- test/scout/network/test_knowledge_base.rb
+- test/scout/network/test_paths.rb
 - test/test_helper.rb
 homepage: http://github.com/mikisvaz/scout-ai
 licenses:
@@ -222,7 +228,7 @@ required_rubygems_version: !ruby/object:Gem::Requirement
     - !ruby/object:Gem::Version
       version: '0'
 requirements: []
-rubygems_version: 3.7.0.dev
+rubygems_version: 3.7.2
 specification_version: 4
 summary: AI gear for scouts
 test_files: []