mementus 0.2.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 6ed27745c387e4e0efed9720fed30792a3535e45
-  data.tar.gz: 34ce7eccc734f7485125cf3568fb56f52cc0ffbc
+  metadata.gz: a4a89d482eebdd901269f7a93501a22883a3b084
+  data.tar.gz: 571323b9865fb729fc2a8b63e70d727a5108a90f
 SHA512:
-  metadata.gz: 74239d3b58b80ec179dd5dfd7f7013ed42ec23e616f9050da61c1ce6f580aa1256d4ae4958b4280fa1030c9ccfa58638da89971bd44ef3ce0604f82fb2a1c8c5
-  data.tar.gz: 68306108232cae430a41f3e51f7df7d14c272c0fa1bc1cc9e5e226bc237c1c00e1cd1c4457053138194bffe7eab15fada9ee1ca1fc32a9f2925ef7b20c6d78d1
+  metadata.gz: 8182797b06e55efe935305ec6067007ae569adbf90ceac25aaf2b990a74495bf64597efd023f24be7d8f6e5ce8bfb59b77da8de666454a7f3f5fdb018c49c42d
+  data.tar.gz: 9151ad44dbcc2570de6e58d60e5e967de8c4ce1ae9441795a63f37ae8bac8ca64cbf1596a15d5a277635fd15b135ae08f3596e4253c5297dcedc63e502bec6da

data/.travis.yml

@@ -1,10 +1,9 @@
 language: ruby
-
-cache: bundler
-
 rvm:
   - 2.1
-  - 2.0
-  - 1.9.3
-
-script: bundle exec rspec
+  - 2.2
+  - rbx-2
+  - jruby-9
+  - jruby-head
+before_install: gem install bundler -v 1.10.6
+script: bundle exec rspec

data/lib/mementus/breadth_first_search.rb

@@ -0,0 +1,28 @@
+module Mementus
+  class BreadthFirstSearch
+    def initialize(graph, start)
+      @graph = graph
+      @start = start
+      @visited = { @start => true }
+      @queue = []
+    end
+
+    def each(&block)
+      visit(NodeProxy.new(@start, graph), &block)
+    end
+
+    private
+
+    def visit(node, &block)
+      @queue << node.adjacent
+
+      while next_node = @queue.shift
+        next if @visited[next_node]
+
+        @visited[next_node] = true
+        block.call(next_node)
+        @queue << next_node.adjacent
+      end
+    end
+  end
+end

data/lib/mementus/depth_first_search.rb

@@ -0,0 +1,25 @@
+module Mementus
+  class DepthFirstSearch
+    def initialize(graph, start)
+      @graph = graph
+      @start = start
+      @visited = { @start => true }
+    end
+
+    def each(&block)
+      visit(NodeProxy.new(@start, @graph), &block)
+    end
+
+    private
+
+    def visit(node, &block)
+      node.each_adjacent do |adjacent|
+        next if @visited[adjacent]
+
+        @visited[adjacent] = true
+        block.call(adjacent)
+        visit(NodeProxy.new(adjacent, @graph), &block)
+      end
+    end
+  end
+end

data/lib/mementus/node_proxy.rb

@@ -0,0 +1,26 @@
+module Mementus
+  class NodeProxy
+    def initialize(node, graph)
+      @node = node
+      @graph = graph
+    end
+
+    def id
+      @node.id
+    end
+
+    def label
+      @node.label
+    end
+
+    def each_adjacent(&block)
+      @graph.each_adjacent(@node, &block)
+    end
+
+    def adjacent
+      each_adjacent.to_a.map do |node|
+        self.class.new(node, @graph)
+      end
+    end
+  end
+end

data/lib/mementus/version.rb

@@ -1,3 +1,3 @@
 module Mementus
-  VERSION = "0.2.0"
+  VERSION = '0.2.1'.freeze
 end

data/lib/mementus.rb

@@ -1,3 +1,6 @@
 require_relative 'mementus/graph'
 require_relative 'mementus/node'
 require_relative 'mementus/edge'
+require_relative 'mementus/node_proxy'
+require_relative 'mementus/depth_first_search'
+require_relative 'mementus/breadth_first_search'

data/spec/node_proxy_spec.rb

@@ -0,0 +1,15 @@
+require 'spec_helper'
+
+describe Mementus::NodeProxy do
+  specify '#new' do
+    graph = Mementus::Graph.new
+    node1 = Mementus::Node.new(1, :node)
+    node2 = Mementus::Node.new(2, :node)
+    node3 = Mementus::Node.new(3, :node)
+    graph.add_edge(Mementus::Edge.new(node1, node2))
+    graph.add_edge(Mementus::Edge.new(node1, node3))
+
+    node_proxy = Mementus::NodeProxy.new(node1, graph)
+    expect(node_proxy.adjacent.map { |node| node.id}).to eq([node2.id, node3.id])
+  end
+end

data/spec/traversal_spec.rb

@@ -0,0 +1,39 @@
+require 'spec_helper'
+
+describe 'Basic graph traversals' do
+  let(:start) do
+    Mementus::Node.new(1, :node)
+  end
+
+  let(:graph) do
+    graph = Mementus::Graph.new
+    graph.add_edge(Mementus::Edge.new(start, Mementus::Node.new(2, :node)))
+    graph.add_edge(Mementus::Edge.new(Mementus::Node.new(2, :node), Mementus::Node.new(3, :node)))
+    graph.add_edge(Mementus::Edge.new(Mementus::Node.new(2, :node), Mementus::Node.new(5, :node)))
+    graph.add_edge(Mementus::Edge.new(Mementus::Node.new(1, :node), Mementus::Node.new(6, :node)))
+    graph.add_edge(Mementus::Edge.new(Mementus::Node.new(2, :node), Mementus::Node.new(7, :node)))
+    graph.add_edge(Mementus::Edge.new(Mementus::Node.new(7, :node), Mementus::Node.new(8, :node)))
+    graph.add_edge(Mementus::Edge.new(Mementus::Node.new(5, :node), Mementus::Node.new(9, :node)))
+    graph
+  end
+
+  specify 'DepthFirstSearch#each' do
+    traversal = Mementus::DepthFirstSearch.new(graph, start)
+
+    expected = [1,2,3,5,9,7,8,6]
+    index = 0
+    traversal.each do |node|
+      expect(node.id).to eq(expected[index + 1])
+    end
+  end
+
+  specify 'BreadthFirstSearch#each' do
+    traversal = Mementus::DepthFirstSearch.new(graph, start)
+
+    expected = [1,2,6,3,5,7,9,8]
+    index = 0
+    traversal.each do |node|
+      expect(node.id).to eq(expected[index + 1])
+    end
+  end
+end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: mementus
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.1
 platform: ruby
 authors:
 - maetl
@@ -67,15 +67,20 @@ files:
 - Rakefile
 - lib/mementus.rb
 - lib/mementus/adapter.rb
+- lib/mementus/breadth_first_search.rb
+- lib/mementus/depth_first_search.rb
 - lib/mementus/edge.rb
 - lib/mementus/graph.rb
 - lib/mementus/model.rb
 - lib/mementus/node.rb
+- lib/mementus/node_proxy.rb
 - lib/mementus/relation.rb
 - lib/mementus/version.rb
 - mementus.gemspec
 - spec/graph_spec.rb
+- spec/node_proxy_spec.rb
 - spec/spec_helper.rb
+- spec/traversal_spec.rb
 homepage: https://github.com/maetl/mementus
 licenses:
 - MIT
@@ -102,5 +107,7 @@ specification_version: 4
 summary: In-memory data model
 test_files:
 - spec/graph_spec.rb
+- spec/node_proxy_spec.rb
 - spec/spec_helper.rb
+- spec/traversal_spec.rb
 has_rdoc: