hierarchical_graph 1.0.0 → 1.0.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 9a3224dabda32c9f8a4b7ba2daff315a9c36282362acc8b509f703039e34d0cb
-  data.tar.gz: 8a290679d5683ce0566d6433c829684961fa440581c49ccedec02c27c22839f8
+  metadata.gz: 501ca870e56fee37c942917a8471f238c82cac2c84a9eecb06ae4fb9440f0f9a
+  data.tar.gz: b5a59a609d92864e3132bb0f5d86154b1ec082f3697fdf97f07d02c2ebb37302
 SHA512:
-  metadata.gz: 0b5540cc426ca14fbd4ca929442ee493fcb6396665e801c6755dafaa01bc17ff03010d10d80cfe75354e0345b60d01b57f68575a17a08a3904463825aad54947
-  data.tar.gz: 52d18fa51a604a5b8f3724e604d430ef286c93bec32a1e7248d40b4808d2a50d6a7abb23bc5f4c6ba4bf81ef7c86ef73e3264a34c02f2dd0961e7e3888faabee
+  metadata.gz: 20ab16944cc5455d6f62278e291426d4e10c66f21e4be0cc6c0491cdd540872163a224bdfd9bf8d161f65b59e7a32bec7a8311aa4ff5ad3970248761d035eca4
+  data.tar.gz: 43f9f1e3cbe70fe4f2df57f326fa116f44c7c0d928025aae98c4dd00fea12c1be0554cd5d20432e983ea914459ef83a7be9238ad73675bf563b39ae8cea685d1

data/lib/hierarchical_graph.rb

@@ -8,8 +8,8 @@ class HierarchicalGraph
 
   def initialize
     @nodes = {}
-    @parent_to_children = Hash.new { |h,k| h[k] = Set.new }
-    @child_to_parents = Hash.new { |h,k| h[k] = Set.new }
+    @parent_to_children = {}
+    @child_to_parents = {}
     @ancestors_cache = {}
     @descendants_cache = {}
   end
@@ -32,6 +32,8 @@ class HierarchicalGraph
 
   def add_node(id, attributes={})
     clear_cache
+    parent_to_children[id] = Set.new
+    child_to_parents[id] = Set.new
     nodes[id] = Node.new self, id, attributes
   end
 
@@ -52,11 +54,11 @@ class HierarchicalGraph
 
   def add_relation(parent_id:, child_id:)
     validate! parent_id, child_id
-    
+
     clear_cache
     parent_to_children[parent_id] << child_id
     child_to_parents[child_id] << parent_id
-    
+
     nil
   end
 
@@ -66,7 +68,7 @@ class HierarchicalGraph
     clear_cache
     parent_to_children[parent_id].delete child_id
     child_to_parents[child_id].delete parent_id
-    
+
     nil
   end
 

data/lib/hierarchical_graph/node.rb

@@ -45,6 +45,6 @@ class HierarchicalGraph
     private
 
     attr_reader :graph
- 
+
   end
 end

data/lib/hierarchical_graph/version.rb

@@ -1,3 +1,3 @@
 class HierarchicalGraph
-  VERSION = '1.0.0'
+  VERSION = '1.0.1'
 end

data/spec/hierarchical_graph_spec.rb

@@ -10,7 +10,7 @@ describe HierarchicalGraph do
 
   it 'Add node' do
     graph = HierarchicalGraph.new
-    
+
     node = graph.add_node 1
 
     graph[1].must_equal node
@@ -20,7 +20,7 @@ describe HierarchicalGraph do
   it 'Remove node' do
     graph = HierarchicalGraph.new
     graph.add_node 1
-    
+
     graph.remove_node 1
 
     graph[1].must_be_nil
@@ -29,7 +29,7 @@ describe HierarchicalGraph do
 
   it 'Remove invalid node' do
     graph = HierarchicalGraph.new
-    
+
     error = proc { graph.remove_node 1 }.must_raise RuntimeError
     error.message.must_equal "Invalid nodes: 1"
   end
@@ -38,7 +38,7 @@ describe HierarchicalGraph do
     graph = HierarchicalGraph.new
     graph.add_node 1
     graph.add_node 2
-    
+
     graph.add_relation parent_id: 1, child_id: 2
 
     graph.parents_of(1).must_be_empty
@@ -50,7 +50,7 @@ describe HierarchicalGraph do
 
   it 'Add relation with invalid nodes' do
     graph = HierarchicalGraph.new
-    
+
     error = proc { graph.add_relation parent_id: 1, child_id: 2 }.must_raise RuntimeError
     error.message.must_equal "Invalid nodes: 1, 2"
   end
@@ -70,7 +70,7 @@ describe HierarchicalGraph do
 
   it 'Remove relation in complex graph' do
     graph = HierarchicalGraph.new
-    
+
     graph.add_node 0
     graph.add_node 1
     graph.add_node 2
@@ -223,6 +223,18 @@ describe HierarchicalGraph do
     graph.tsort.map(&:id).must_equal [2, 4, 3, 1]
   end
 
+  it 'Marshal' do
+    graph = HierarchicalGraph.new
+    graph.add_node 1
+    graph.add_node 2
+    graph.add_relation parent_id: 1, child_id: 2
+
+    dump = Marshal.dump graph
+    loaded = Marshal.load dump
+
+    loaded.to_s.must_equal graph.to_s
+  end
+
   it 'To string' do
     graph = HierarchicalGraph.new
     graph.add_node 1

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: hierarchical_graph
 version: !ruby/object:Gem::Version
-  version: 1.0.0
+  version: 1.0.1
 platform: ruby
 authors:
 - Gabriel Naiman
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-04-12 00:00:00.000000000 Z
+date: 2020-10-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake