Fix Cell#to_cell method

project-eutopia · project-eutopia · commit 6b3c332ffb8b · 2021-05-21T17:08:17.000+09:00
Previously this could cause there to be Cell&lt;Cell&lt;Node&gt;&gt; objects created
when calling #to_cell on a Cell.
diff --git a/lib/keisan/ast/cell.rb b/lib/keisan/ast/cell.rb
@@ -66,7 +66,7 @@ def replace(variable, replacement)
       end
 
       def to_cell
-        self.class.new(node.to_cell)
+        self.class.new(node.to_node)
       end
 
       def to_s
diff --git a/spec/keisan/ast/cell_spec.rb b/spec/keisan/ast/cell_spec.rb
@@ -31,4 +31,12 @@
       expect(cell.to_node).to eq node
     end
   end
+
+  describe "#to_cell" do
+    it "wraps the node in a new cell" do
+      new_cell = cell.to_cell
+      expect(cell).not_to eq new_cell
+      expect(cell.node).to eq new_cell.node
+    end
+  end
 end
