Improve coverage

project-eutopia · Nov 23, 2018 · a5189c9 · a5189c9
1 parent efa385b
commit a5189c9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/spec/keisan/default_functions_spec.rb b/spec/keisan/default_functions_spec.rb
@@ -238,7 +238,7 @@
     describe "continue" do
       it "skips to the next iteration of the loop" do
         calculator = Keisan::Calculator.new
-        expect(calculator.evaluate("x = 0; sum = 0; while(x < 10, x += 1; if (x % 2 == 0, continue); sum += x**2); sum")).to eq(
+        expect(calculator.simplify("x = 0; sum = 0; while(x < 10, x += 1; if (x % 2 == 0, continue); sum += x**2); sum").value).to eq(
           1 + 3**2 + 5**2 + 7**2 + 9**2
         )
       end

diff --git a/spec/keisan/functions/while_spec.rb b/spec/keisan/functions/while_spec.rb
@@ -26,4 +26,10 @@
     expect(calculator.evaluate("[1,2,3].includes(2)")).to eq true
     expect(calculator.evaluate("[1,2,3].includes(4)")).to eq false
   end
+
+  it "must have boolean in condition expression" do
+    calculator = Keisan::Calculator.new
+    calculator.evaluate("x = 0")
+    expect{calculator.evaluate("while(!x, x = x + 1)")}.to raise_error(Keisan::Exceptions::InvalidFunctionError)
+  end
 end