update tests

viktordanov · Sep 26, 2022 · 4566d72 · 4566d72
1 parent ae23507
commit 4566d72
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/cmd/abacus-cli/main_test.go b/cmd/abacus-cli/main_test.go
@@ -30,6 +30,11 @@ func Test_evaluateExpression(t *testing.T) {
 		{expr: "2", want: abacus.NewNumber(2), visitor: visitor},
 		{expr: "2.139487526 + 9.4777777", want: abacus.NewNumber(11.617265226), visitor: visitor},
 	}
+	expressionTests := testArgs{
+		{expr: "2", want: abacus.NewNumber(2), visitor: visitor},
+		{expr: "0.2", want: abacus.NewNumber(0.2), visitor: visitor},
+		{expr: ".2", want: abacus.NewNumber(0.2), visitor: visitor},
+	}
 	multiplicationTests := testArgs{
 		{expr: "2*2", want: abacus.NewNumber(4), visitor: visitor},
 		{expr: "2*2*2", want: abacus.NewNumber(8), visitor: visitor},
@@ -70,6 +75,7 @@ func Test_evaluateExpression(t *testing.T) {
 	}
 
 	runTestSuite(t, "Addition & Subtraction", additionTests)
+	runTestSuite(t, "Expressions", expressionTests)
 	runTestSuite(t, "Multiplication & Division", multiplicationTests)
 	runTestSuite(t, "Exponentiation", exponentiationTests)
 	runTestSuite(t, "Variables", variableTests)