mementus 0.2.2 → 0.2.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 26dd9d077d7ad9d1747b4071a91db233131dbba3
-  data.tar.gz: d04e4c35b6eca0c45640cb06ba0136ab08b2453a
+  metadata.gz: 92652ba97da7674692fac49d6ea139aeb9c680d5
+  data.tar.gz: 5ea49d59abd929eb237956e2dea519e1300e94c6
 SHA512:
-  metadata.gz: 9cf713cbafc0edd0df14e241a7c7b328948eeac489146ce96d3849fd52ffb8a75fb9f124b78beaec5828d3485a5b09099c9452bbc60a26637b67f8f30edea528
-  data.tar.gz: 6541ecd1725179928a9dc9c6cdafe0df633222da04bff157e5b63dc3aa0b30699ce84bdcc6a14e75a0c6afef95d6b4ea9e7bfb59beca86f2954e67f9e524c13b
+  metadata.gz: 6bf3f09c02410fbc0070d40d0016d5161f57cecf975a660ccedebcb0d10fdebdf11e01cf63e350d1fc9d33fface9ebc81d9b64002dba5911221d4dd1a6f2b3c5
+  data.tar.gz: c36019eb7529244bf07b4f5c1654c87e5d55c6e675e16bd82fcdce6fccb12a328d78595170b3dc698b4a41a37967c4b437c0d19c945a0bba5e3b3d0c50cb5589

data/lib/mementus/edge.rb

@@ -1,10 +1,31 @@
 module Mementus
   class Edge
-    attr_reader :from, :to
+    attr_reader :from, :to, :label
 
-    def initialize(from, to)
+    def initialize(from=nil, to=nil, label=:edge)
       @from = from
       @to = to
+      @label = label
+    end
+
+    def from=(node, label=:node)
+      if node.is_a?(Node)
+        @from = node
+      else
+        @from = Node.new(node, label)
+      end
+    end
+
+    def to=(node)
+      if node.is_a?(Node)
+        @to = node
+      else
+        @to = Node.new(node, label)
+      end
+    end
+
+    def label=(label)
+      @label = label
     end
 
     def nodes

data/lib/mementus/graph.rb

@@ -25,6 +25,18 @@ module Mementus
       @index[node] ||= Set.new
     end
 
+    def create_edge(&block)
+      edge = Edge.new
+      yield edge
+      add_edge(edge)
+    end
+
+    def create_node(&block)
+      node = Node.new
+      yield node
+      add_node(node)
+    end
+
     def add_edge(edge)
       add_node(edge.from)
       add_node(edge.to)

data/lib/mementus/node.rb

@@ -2,9 +2,17 @@ module Mementus
   class Node
     attr_reader :id, :label
 
-    def initialize(id, label)
+    def initialize(id=nil, label=:node)
       @id = id
       @label = label
     end
+
+    def id=(id)
+      @id = id
+    end
+
+    def label=(label)
+      @label = label
+    end
   end
 end

data/lib/mementus/version.rb

@@ -1,3 +1,3 @@
 module Mementus
-  VERSION = '0.2.2'.freeze
+  VERSION = '0.2.3'.freeze
 end

data/spec/graph_spec.rb

@@ -16,6 +16,17 @@ describe Mementus::Graph do
     expect(graph.edges_count).to eq(0)
   end
 
+  specify '#create_node' do
+    graph = Mementus::Graph.new
+    graph.create_node do |node|
+      node.id = 1
+      node.label = :vertex
+    end
+
+    expect(graph.nodes_count).to eq(1)
+    expect(graph.edges_count).to eq(0)
+  end
+
   specify '#add_edge' do
     graph = Mementus::Graph.new
     graph.add_edge(Mementus::Edge.new(Mementus::Node.new(1, :node), Mementus::Node.new(2, :node)))
@@ -24,6 +35,18 @@ describe Mementus::Graph do
     expect(graph.edges_count).to eq(1)
   end
 
+  specify '#create_edge' do
+    graph = Mementus::Graph.new
+    graph.create_edge do |edge|
+      edge.label = :relationship
+      edge.from = "A"
+      edge.to = "B"
+    end
+
+    expect(graph.nodes_count).to eq(2)
+    expect(graph.edges_count).to eq(1)
+  end
+
   specify '#has_node?' do
     graph = Mementus::Graph.new
     node = Mementus::Node.new(1, :node)

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: mementus
 version: !ruby/object:Gem::Version
-  version: 0.2.2
+  version: 0.2.3
 platform: ruby
 authors:
 - maetl
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2016-03-29 00:00:00.000000000 Z
+date: 2016-04-11 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler