Add tests for convenience rewrite function

davedelong · Apr 29, 2017 · 096908e · 096908e
1 parent 2a4db28
commit 096908e
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/MathParserTests/RewriterTests.swift b/MathParserTests/RewriterTests.swift
@@ -18,6 +18,10 @@ func TestRewrite(_ original: String, expected: String, substitutions: Substituti
     let rewritten = rewriter.rewriteExpression(originalE, substitutions: substitutions, evaluator: evaluator)
 
     XCTAssertEqual(rewritten, expectedE, file: file, line: line)
+
+    // Also test that the convenience functions produces the same result.
+    let convenienceRewritten = originalE.rewrite(substitutions, rewriter: rewriter, evaluator: evaluator)
+    XCTAssertEqual(convenienceRewritten, rewritten)
 }
 
 class RewriterTests: XCTestCase {