Files
swiftui-math/SwiftMath/Tests/SwiftMathTests/MTMathListBuilderTests.swift
Guille Gonzalez f805b3adf5 Clean slate
2025-12-31 12:40:11 +01:00

2541 lines
101 KiB
Swift

import XCTest
@testable import SwiftUIMath
//
// MathRenderSwiftTests.swift
// MathRenderSwiftTests
//
// Created by Mike Griebling on 2023-01-02.
//
final class MTMathListBuilderTests: XCTestCase {
func checkAtomTypes(_ list:MTMathList?, types:[MTMathAtomType], desc:String) {
if let list = list {
XCTAssertEqual(list.atoms.count, types.count, desc)
for i in 0..<list.atoms.count {
let atom = list.atoms[i]
XCTAssertNotNil(atom, desc)
XCTAssertEqual(atom.type, types[i], desc)
}
} else {
XCTAssert(types.count == 0, "MathList should have no atoms!")
}
}
// override func setUpWithError() throws {
// // Put setup code here. This method is called before the invocation of each test method in the class.
// }
//
// override func tearDownWithError() throws {
// // Put teardown code here. This method is called after the invocation of each test method in the class.
// }
struct TestRecord {
let build : String
let atomType : [MTMathAtomType]
let types : [MTMathAtomType]
let extra : [MTMathAtomType]
let result : String
init(build: String, atomType: [MTMathAtomType], types: [MTMathAtomType], extra: [MTMathAtomType] = [MTMathAtomType](), result: String) {
self.build = build
self.atomType = atomType
self.types = types
self.extra = extra
self.result = result
}
}
func getTestData() -> [TestRecord] {
[
TestRecord(build: "x", atomType: [.variable ], types: [], result: "x"),
TestRecord(build: "1", atomType: [.number ] , types: [], result: "1"),
TestRecord(build: "*", atomType: [.binaryOperator ] ,types: [], result:"*"),
TestRecord(build: "+", atomType: [.binaryOperator ], types: [], result:"+"),
TestRecord(build: ".", atomType: [.number ], types: [], result:"."),
TestRecord(build: "(", atomType: [.open ], types: [], result:"(" ),
TestRecord(build: ")", atomType: [.close ], types: [], result:")"),
TestRecord(build: ",", atomType: [.punctuation], types: [], result:","),
TestRecord(build: "!", atomType: [.close], types: [], result:"!"),
TestRecord(build: "=", atomType: [.relation], types: [], result:"="),
TestRecord(build: "x+2", atomType: [.variable, .binaryOperator, .number ], types: [], result:"x+2"),
// spaces are ignored
TestRecord(build: "(2.3 * 8)", atomType: [.open, .number, .number, .number, .binaryOperator, .number , .close ], types: [], result:"(2.3*8)"),
// braces are just for grouping
TestRecord(build: "5{3+4}", atomType: [.number, .number, .binaryOperator, .number], types: [], result:"53+4"),
// commands
TestRecord(build: "\\pi+\\theta\\geq 3",atomType: [.variable, .binaryOperator, .variable, .relation, .number], types: [], result:"\\pi +\\theta \\geq 3"),
// aliases
TestRecord(build: "\\pi\\ne 5 \\land 3", atomType: [.variable, .relation, .number, .binaryOperator, .number], types: [], result:"\\pi \\neq 5\\wedge 3"),
// control space
TestRecord(build: "x \\ y", atomType: [ .variable, .ordinary, .variable], types: [], result:"x\\ y"),
// spacing
TestRecord(build: "x \\quad y \\; z \\! q", atomType: [ .variable, .space, .variable,.space, .variable, .space, .variable], types: [], result:"x\\quad y\\; z\\! q")
]
}
func getTestDataSuperScript() -> [TestRecord] {
[
TestRecord(build: "x^2", atomType: [.variable], types: [.number], result: "x^{2}"),
TestRecord(build: "x^23", atomType: [ .variable, .number ], types: [ .number ], result: "x^{2}3"),
TestRecord(build: "x^{23}", atomType: [ .variable ], types: [ .number, .number ], result: "x^{23}"),
TestRecord(build: "x^2^3", atomType: [ .variable, .ordinary ], types: [ .number ], result: "x^{2}{}^{3}" ),
TestRecord(build: "x^{2^3}", atomType: [ .variable ], types: [ .number], extra: [ .number ], result: "x^{2^{3}}"),
TestRecord(build: "x^{^2*}", atomType: [ .variable ], types: [ .ordinary, .binaryOperator], extra:[ .number ], result:"x^{{}^{2}*}"),
TestRecord(build: "^2", atomType: [ .ordinary], types: [ .number ], result: "{}^{2}"),
TestRecord(build: "{}^2", atomType: [ .ordinary], types: [ .number ], result: "{}^{2}"),
TestRecord(build: "x^^2", atomType: [ .variable, .ordinary ], types: [ ], result: "x^{}{}^{2}"),
TestRecord(build: "5{x}^2", atomType: [ .number, .variable], types: [ ], result: "5x^{2}"),
]
}
func getTestDataSubScript() -> [TestRecord] {
[
TestRecord(build: "x_2", atomType: [.variable], types: [.number], result: "x_{2}"),
TestRecord(build: "x_23", atomType: [ .variable, .number ], types: [ .number ], result: "x_{2}3"),
TestRecord(build: "x_{23}", atomType: [ .variable ], types: [ .number, .number ], result: "x_{23}"),
TestRecord(build: "x_2_3", atomType: [ .variable, .ordinary ], types: [ .number ], result: "x_{2}{}_{3}" ),
TestRecord(build: "x_{2_3}", atomType: [ .variable ], types: [ .number], extra: [ .number ], result: "x_{2_{3}}"),
TestRecord(build: "x_{_2*}", atomType: [ .variable ], types: [ .ordinary, .binaryOperator], extra:[ .number ], result:"x_{{}_{2}*}"),
TestRecord(build: "_2", atomType: [ .ordinary], types: [ .number ], result: "{}_{2}"),
TestRecord(build: "{}_2", atomType: [ .ordinary], types: [ .number ], result: "{}_{2}"),
TestRecord(build: "x__2", atomType: [ .variable, .ordinary ], types: [ ], result: "x_{}{}_{2}"),
TestRecord(build: "5{x}_2", atomType: [ .number, .variable], types: [ ], result: "5x_{2}"),
]
}
func getTestDataSuperSubScript() -> [TestRecord] {
[
TestRecord(build: "x_2^*", atomType: [.variable], types: [.number], extra: [.binaryOperator], result: "x^{*}_{2}"),
TestRecord(build: "x^*_2", atomType: [.variable], types: [.number], extra: [.binaryOperator], result: "x^{*}_{2}"),
TestRecord(build: "x_^*", atomType: [.variable], types: [ ], extra: [.binaryOperator], result: "x^{*}_{}"),
TestRecord(build: "x^_2", atomType: [.variable], types: [.number], result: "x^{}_{2}"),
TestRecord(build: "x_{2^*}", atomType: [.variable], types: [.number], result: "x_{2^{*}}"),
TestRecord(build: "x^{*_2}", atomType: [.variable], types: [ ], extra: [.binaryOperator], result: "x^{*_{2}}"),
TestRecord(build: "_2^*", atomType: [.ordinary], types: [.number], extra: [.binaryOperator], result: "{}^{*}_{2}")
]
}
struct TestRecord2 {
let build : String
let type1 : [MTMathAtomType]
let number : Int
let type2 : [MTMathAtomType]
let left : String
let right : String
let result : String
}
func getTestDataLeftRight() -> [TestRecord2] {
[
TestRecord2(build: "\\left( 2 \\right)", type1: [ .inner ], number: 0, type2: [ .number], left: "(", right: ")", result: "\\left( 2\\right) "),
// spacing
TestRecord2(build: "\\left ( 2 \\right )", type1: [ .inner ], number: 0, type2: [ .number], left: "(", right: ")", result: "\\left( 2\\right) "),
// commands
TestRecord2(build: "\\left\\{ 2 \\right\\}", type1: [ .inner ], number: 0, type2: [ .number], left: "{", right: "}", result: "\\left\\{ 2\\right\\} "),
// complex commands
TestRecord2(build: "\\left\\langle x \\right\\rangle", type1: [ .inner ], number: 0, type2: [ .variable], left: "\u{2329}", right: "\u{232A}", result: "\\left< x\\right> "),
// bars
TestRecord2(build: "\\left| x \\right\\|", type1: [ .inner ], number: 0, type2: [ .variable], left: "|", right: "\u{2016}", result: "\\left| x\\right\\| "),
// inner in between
TestRecord2(build: "5 + \\left( 2 \\right) - 2", type1: [ .number, .binaryOperator, .inner, .binaryOperator, .number ], number: 2, type2: [ .number], left: "(", right: ")", result: "5+\\left( 2\\right) -2"),
// long inner
TestRecord2(build: "\\left( 2 + \\frac12\\right)", type1: [ .inner ], number: 0, type2: [ .number, .binaryOperator, .fraction], left: "(", right: ")", result: "\\left( 2+\\frac{1}{2}\\right) "),
// nested
TestRecord2(build: "\\left[ 2 + \\left|\\frac{-x}{2}\\right| \\right]", type1: [ .inner ], number: 0, type2: [ .number, .binaryOperator, .inner], left: "[", right: "]", result: "\\left[ 2+\\left| \\frac{-x}{2}\\right| \\right] "),
// With scripts
TestRecord2(build: "\\left( 2 \\right)^2", type1: [ .inner ], number: 0, type2: [ .number], left: "(", right: ")", result: "\\left( 2\\right) ^{2}"),
// Scripts on left
TestRecord2(build: "\\left(^2 \\right )", type1: [ .inner], number: 0, type2: [ .ordinary], left: "(", right: ")", result: "\\left( {}^{2}\\right) "),
// Dot
TestRecord2(build: "\\left( 2 \\right.", type1: [ .inner], number: 0, type2: [ .number], left: "(", right: "", result: "\\left( 2\\right. ")
]
}
func getTestDataParseErrors() -> [(String, MTParseErrors)] {
return [
("}a", .mismatchBraces),
("\\notacommand", .invalidCommand),
("\\sqrt[5+3", .characterNotFound),
("{5+3", .mismatchBraces),
("5+3}", .mismatchBraces),
("{1+\\frac{3+2", .mismatchBraces),
("1+\\left", .missingDelimiter),
("\\left(\\frac12\\right", .missingDelimiter),
("\\left 5 + 3 \\right)", .invalidDelimiter),
("\\left(\\frac12\\right + 3", .invalidDelimiter),
("\\left\\lmoustache 5 + 3 \\right)", .invalidDelimiter),
("\\left(\\frac12\\right\\rmoustache + 3", .invalidDelimiter),
("5 + 3 \\right)", .missingLeft),
("\\left(\\frac12", .missingRight),
("\\left(5 + \\left| \\frac12 \\right)", .missingRight),
("5+ \\left|\\frac12\\right| \\right)", .missingLeft),
("\\begin matrix \\end matrix", .characterNotFound), // missing {
("\\begin", .characterNotFound), // missing {
("\\begin{", .characterNotFound), // missing }
("\\begin{matrix parens}", .characterNotFound), // missing } (no spaces in env)
("\\begin{matrix} x", .missingEnd),
("\\begin{matrix} x \\end", .characterNotFound), // missing {
("\\begin{matrix} x \\end + 3", .characterNotFound), // missing {
("\\begin{matrix} x \\end{", .characterNotFound), // missing }
("\\begin{matrix} x \\end{matrix + 3", .characterNotFound), // missing }
("\\begin{matrix} x \\end{pmatrix}", .invalidEnv),
("x \\end{matrix}", .missingBegin),
("\\begin{notanenv} x \\end{notanenv}", .invalidEnv),
("\\begin{matrix} \\notacommand \\end{matrix}", .invalidCommand),
("\\begin{displaylines} x & y \\end{displaylines}", .invalidNumColumns),
("\\begin{eqalign} x \\end{eqalign}", .invalidNumColumns),
("\\nolimits", .invalidLimits),
("\\frac\\limits{1}{2}", .invalidLimits),
("&\\begin", .characterNotFound),
("x & y \\\\ z & w \\end{matrix}", .invalidEnv)
]
}
func testBuilder() throws {
let data = getTestData()
for testCase in data {
let str = testCase.build
var error : NSError? = nil
let list = MTMathListBuilder.build(fromString: str, error: &error)
XCTAssertNil(error)
let desc = "Error for string:\(str)"
let atomTypes = testCase.atomType
self.checkAtomTypes(list, types:atomTypes, desc:desc)
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, testCase.result, desc)
}
}
func testSuperScript() throws {
let data = getTestDataSuperScript()
for testCase in data {
let str = testCase.build
var error : NSError? = nil
let list = MTMathListBuilder.build(fromString: str, error:&error)
XCTAssertNil(error)
let desc = "Error for string:\(str)"
let atomTypes = testCase.atomType
checkAtomTypes(list, types:atomTypes, desc:desc)
// get the first atom
let first = list!.atoms[0]
// check it's superscript
let types = testCase.types
if types.count > 0 {
XCTAssertNotNil(first.superScript, desc)
}
let superlist = first.superScript
checkAtomTypes(superlist, types:types, desc:desc)
if !testCase.extra.isEmpty {
// one more level
let superFirst = superlist!.atoms[0]
let supersuperList = superFirst.superScript
checkAtomTypes(supersuperList, types:testCase.extra, desc:desc)
}
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, testCase.result, desc)
}
}
func testSubScript() throws {
let data = getTestDataSubScript()
for testCase in data {
let str = testCase.build
var error : NSError? = nil
let list = MTMathListBuilder.build(fromString: str, error:&error)
XCTAssertNil(error)
let desc = "Error for string:\(str)"
let atomTypes = testCase.atomType
checkAtomTypes(list, types:atomTypes, desc:desc)
// get the first atom
let first = list!.atoms[0]
// check it's superscript
let types = testCase.types
if (types.count > 0) {
XCTAssertNotNil(first.subScript, desc);
}
let sublist = first.subScript
checkAtomTypes(sublist, types:types, desc:desc)
if !testCase.extra.isEmpty {
// one more level
let subFirst = sublist!.atoms[0]
let subsubList = subFirst.subScript
checkAtomTypes(subsubList, types:testCase.extra, desc:desc)
}
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, testCase.result, desc)
}
}
func testSuperSubScript() throws {
let data = getTestDataSuperSubScript()
for testCase in data {
let str = testCase.build
var error : NSError? = nil
let list = MTMathListBuilder.build(fromString: str, error:&error)
XCTAssertNil(error)
let desc = "Error for string:\(str)"
let atomTypes = testCase.atomType
checkAtomTypes(list, types:atomTypes, desc:desc)
// get the first atom
let first = list!.atoms[0]
// check its subscript
let sub = testCase.types
if sub.count > 0 {
XCTAssertNotNil(first.subScript, desc)
let sublist = first.subScript
checkAtomTypes(sublist, types: sub, desc: desc)
}
let sup = testCase.extra
if sup.count > 0 {
XCTAssertNotNil(first.superScript, desc)
let sublist = first.superScript
checkAtomTypes(sublist, types: sup, desc: desc)
}
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, testCase.result, desc)
}
}
func testSymbols() throws {
let str = "5\\times3^{2\\div2}";
let list = MTMathListBuilder.build(fromString: str)!
let desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc)
XCTAssertEqual((list.atoms.count), 3, desc)
var atom = list.atoms[0];
XCTAssertEqual(atom.type, .number, desc)
XCTAssertEqual(atom.nucleus, "5", desc)
atom = list.atoms[1];
XCTAssertEqual(atom.type, .binaryOperator, desc)
XCTAssertEqual(atom.nucleus, "\u{00D7}", desc)
atom = list.atoms[2];
XCTAssertEqual(atom.type, .number, desc)
XCTAssertEqual(atom.nucleus, "3", desc)
// super script
let superList = atom.superScript!
XCTAssertNotNil(superList, desc)
XCTAssertEqual((superList.atoms.count), 3, desc)
atom = superList.atoms[0];
XCTAssertEqual(atom.type, .number, desc)
XCTAssertEqual(atom.nucleus, "2", desc)
atom = superList.atoms[1];
XCTAssertEqual(atom.type, .binaryOperator, desc)
XCTAssertEqual(atom.nucleus, "\u{00F7}", desc)
atom = superList.atoms[2];
XCTAssertEqual(atom.type, .number, desc)
XCTAssertEqual(atom.nucleus, "2", desc)
}
func testFrac() throws {
let str = "\\frac1c";
let list = MTMathListBuilder.build(fromString: str)!
let desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc)
XCTAssertEqual((list.atoms.count), 1, desc)
let frac = list.atoms[0] as! MTFraction
XCTAssertEqual(frac.type, .fraction, desc)
XCTAssertEqual(frac.nucleus, "", desc)
XCTAssertTrue(frac.hasRule);
XCTAssertTrue(frac.rightDelimiter.isEmpty)
XCTAssertTrue(frac.leftDelimiter.isEmpty)
var subList = frac.numerator!
XCTAssertNotNil(subList, desc)
XCTAssertEqual((subList.atoms.count), 1, desc)
var atom = subList.atoms[0];
XCTAssertEqual(atom.type, .number, desc)
XCTAssertEqual(atom.nucleus, "1", desc)
atom = list.atoms[0];
subList = frac.denominator!
XCTAssertNotNil(subList, desc)
XCTAssertEqual((subList.atoms.count), 1, desc)
atom = subList.atoms[0];
XCTAssertEqual(atom.type, .variable, desc)
XCTAssertEqual(atom.nucleus, "c", desc)
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\frac{1}{c}", desc)
}
func testFracInFrac() throws {
let str = "\\frac1\\frac23";
let list = MTMathListBuilder.build(fromString: str)!
let desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc)
XCTAssertEqual((list.atoms.count), 1, desc)
var frac = list.atoms[0] as! MTFraction
XCTAssertEqual(frac.type, .fraction, desc)
XCTAssertEqual(frac.nucleus, "", desc)
XCTAssertTrue(frac.hasRule);
var subList = frac.numerator!
XCTAssertNotNil(subList, desc)
XCTAssertEqual((subList.atoms.count), 1, desc)
var atom = subList.atoms[0];
XCTAssertEqual(atom.type, .number, desc)
XCTAssertEqual(atom.nucleus, "1", desc)
subList = frac.denominator!
XCTAssertNotNil(subList, desc)
XCTAssertEqual((subList.atoms.count), 1, desc)
frac = subList.atoms[0] as! MTFraction
XCTAssertEqual(frac.type, .fraction, desc)
XCTAssertEqual(frac.nucleus, "", desc)
subList = frac.numerator!
XCTAssertNotNil(subList, desc)
XCTAssertEqual((subList.atoms.count), 1, desc)
atom = subList.atoms[0];
XCTAssertEqual(atom.type, .number, desc)
XCTAssertEqual(atom.nucleus, "2", desc)
subList = frac.denominator!
XCTAssertNotNil(subList, desc)
XCTAssertEqual((subList.atoms.count), 1, desc)
atom = subList.atoms[0];
XCTAssertEqual(atom.type, .number, desc)
XCTAssertEqual(atom.nucleus, "3", desc)
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\frac{1}{\\frac{2}{3}}", desc)
}
func testSqrt() throws {
let str = "\\sqrt2";
let list = MTMathListBuilder.build(fromString: str)!
let desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc)
XCTAssertEqual((list.atoms.count), 1, desc)
let rad = list.atoms[0] as! MTRadical
XCTAssertEqual(rad.type, .radical, desc)
XCTAssertEqual(rad.nucleus, "", desc)
let subList = rad.radicand!
XCTAssertNotNil(subList, desc)
XCTAssertEqual((subList.atoms.count), 1, desc)
let atom = subList.atoms[0]
XCTAssertEqual(atom.type, .number, desc)
XCTAssertEqual(atom.nucleus, "2", desc)
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\sqrt{2}", desc)
}
func testSqrtInSqrt() throws {
let str = "\\sqrt\\sqrt2";
let list = MTMathListBuilder.build(fromString: str)!
let desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc)
XCTAssertEqual((list.atoms.count), 1, desc)
var rad = list.atoms[0] as! MTRadical
XCTAssertEqual(rad.type, .radical, desc)
XCTAssertEqual(rad.nucleus, "", desc)
var subList = rad.radicand!
XCTAssertNotNil(subList, desc)
XCTAssertEqual((subList.atoms.count), 1, desc)
rad = subList.atoms[0] as! MTRadical
XCTAssertEqual(rad.type, .radical, desc)
XCTAssertEqual(rad.nucleus, "", desc)
subList = rad.radicand!
XCTAssertNotNil(subList, desc)
XCTAssertEqual((subList.atoms.count), 1, desc)
let atom = subList.atoms[0];
XCTAssertEqual(atom.type, .number, desc)
XCTAssertEqual(atom.nucleus, "2", desc)
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\sqrt{\\sqrt{2}}", desc)
}
func testRad() throws {
let str = "\\sqrt[3]2";
let list = MTMathListBuilder.build(fromString: str)!
XCTAssertNotNil(list);
XCTAssertEqual((list.atoms.count), 1);
let rad = list.atoms[0] as! MTRadical
XCTAssertEqual(rad.type, .radical);
XCTAssertEqual(rad.nucleus, "");
var subList = rad.radicand!
XCTAssertNotNil(subList);
XCTAssertEqual((subList.atoms.count), 1);
var atom = subList.atoms[0];
XCTAssertEqual(atom.type, .number);
XCTAssertEqual(atom.nucleus, "2");
subList = rad.degree!
XCTAssertNotNil(subList);
XCTAssertEqual((subList.atoms.count), 1);
atom = subList.atoms[0];
XCTAssertEqual(atom.type, .number);
XCTAssertEqual(atom.nucleus, "3");
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\sqrt[3]{2}");
}
func testSqrtWithoutRadicand() throws {
let str = "\\sqrt"
let list = try XCTUnwrap(MTMathListBuilder.build(fromString: str))
XCTAssertEqual(list.atoms.count, 1)
let rad = try XCTUnwrap(list.atoms.first as? MTRadical)
XCTAssertEqual(rad.type, .radical)
XCTAssertEqual(rad.nucleus, "")
XCTAssertEqual(rad.radicand?.atoms.isEmpty, true)
XCTAssertNil(rad.degree)
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\sqrt{}")
}
func testSqrtWithDegreeWithoutRadicand() throws {
let str = "\\sqrt[3]"
let list = try XCTUnwrap(MTMathListBuilder.build(fromString: str))
XCTAssertEqual(list.atoms.count, 1)
let rad = try XCTUnwrap(list.atoms.first as? MTRadical)
XCTAssertEqual(rad.type, .radical)
XCTAssertEqual(rad.nucleus, "")
XCTAssertEqual(rad.radicand?.atoms.isEmpty, true)
let subList = try XCTUnwrap(rad.degree)
XCTAssertEqual(subList.atoms.count, 1)
let atom = try XCTUnwrap(subList.atoms.first)
XCTAssertEqual(atom.type, .number)
XCTAssertEqual(atom.nucleus, "3")
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\sqrt[3]{}");
}
func testLeftRight() throws {
let data = getTestDataLeftRight()
for testCase in data {
let str = testCase.build
var error : NSError? = nil
let list = MTMathListBuilder.build(fromString: str, error: &error)!
XCTAssertNotNil(list, str);
XCTAssertNil(error, str);
checkAtomTypes(list, types:testCase.type1, desc:"\(str) outer")
let innerLoc = testCase.number
let inner = list.atoms[innerLoc] as! MTInner
XCTAssertEqual(inner.type, .inner, str);
XCTAssertEqual(inner.nucleus, "", str);
let innerList = inner.innerList!
XCTAssertNotNil(innerList, str);
checkAtomTypes(innerList, types:testCase.type2, desc:"\(str) inner")
XCTAssertNotNil(inner.leftBoundary, str);
XCTAssertEqual(inner.leftBoundary!.type, .boundary, str);
XCTAssertEqual(inner.leftBoundary!.nucleus, testCase.left, str);
XCTAssertNotNil(inner.rightBoundary, str);
XCTAssertEqual(inner.rightBoundary!.type, .boundary, str);
XCTAssertEqual(inner.rightBoundary!.nucleus, testCase.right, str);
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, testCase.result, str);
}
}
func testOver() throws {
let str = "1 \\over c";
let list = MTMathListBuilder.build(fromString:str)!
let desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc);
XCTAssertEqual((list.atoms.count), 1, desc);
let frac = list.atoms[0] as! MTFraction
XCTAssertEqual(frac.type, .fraction, desc);
XCTAssertEqual(frac.nucleus, "", desc);
XCTAssertTrue(frac.hasRule);
XCTAssertTrue(frac.rightDelimiter.isEmpty)
XCTAssertTrue(frac.leftDelimiter.isEmpty)
var subList = frac.numerator!
XCTAssertNotNil(subList, desc);
XCTAssertEqual((subList.atoms.count), 1, desc);
var atom = subList.atoms[0];
XCTAssertEqual(atom.type, .number, desc);
XCTAssertEqual(atom.nucleus, "1", desc);
atom = list.atoms[0];
subList = frac.denominator!
XCTAssertNotNil(subList, desc);
XCTAssertEqual((subList.atoms.count), 1, desc);
atom = subList.atoms[0];
XCTAssertEqual(atom.type, .variable, desc);
XCTAssertEqual(atom.nucleus, "c", desc);
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\frac{1}{c}", desc);
}
func testOverInParens() throws {
let str = "5 + {1 \\over c} + 8";
let list = MTMathListBuilder.build(fromString:str)!
let desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc);
XCTAssertEqual((list.atoms.count), 5, desc);
let types = [MTMathAtomType.number, .binaryOperator, .fraction, .binaryOperator, .number]
self.checkAtomTypes(list, types:types, desc:desc)
let frac = list.atoms[2] as! MTFraction
XCTAssertEqual(frac.type, .fraction, desc);
XCTAssertEqual(frac.nucleus, "", desc);
XCTAssertTrue(frac.hasRule);
XCTAssertTrue(frac.rightDelimiter.isEmpty)
XCTAssertTrue(frac.leftDelimiter.isEmpty)
var subList = frac.numerator!
XCTAssertNotNil(subList, desc);
XCTAssertEqual((subList.atoms.count), 1, desc);
var atom = subList.atoms[0];
XCTAssertEqual(atom.type, .number, desc);
XCTAssertEqual(atom.nucleus, "1", desc);
atom = list.atoms[0];
subList = frac.denominator!
XCTAssertNotNil(subList, desc);
XCTAssertEqual((subList.atoms.count), 1, desc);
atom = subList.atoms[0];
XCTAssertEqual(atom.type, .variable, desc);
XCTAssertEqual(atom.nucleus, "c", desc);
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "5+\\frac{1}{c}+8", desc);
}
func testAtop() throws {
let str = "1 \\atop c";
let list = MTMathListBuilder.build(fromString:str)!
let desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc);
XCTAssertEqual((list.atoms.count), 1, desc);
let frac = list.atoms[0] as! MTFraction
XCTAssertEqual(frac.type, .fraction, desc);
XCTAssertEqual(frac.nucleus, "", desc);
XCTAssertFalse(frac.hasRule);
XCTAssertTrue(frac.rightDelimiter.isEmpty)
XCTAssertTrue(frac.leftDelimiter.isEmpty)
var subList = frac.numerator!
XCTAssertNotNil(subList, desc);
XCTAssertEqual((subList.atoms.count), 1, desc);
var atom = subList.atoms[0];
XCTAssertEqual(atom.type, .number, desc);
XCTAssertEqual(atom.nucleus, "1", desc);
atom = list.atoms[0];
subList = frac.denominator!
XCTAssertNotNil(subList, desc);
XCTAssertEqual((subList.atoms.count), 1, desc);
atom = subList.atoms[0];
XCTAssertEqual(atom.type, .variable, desc);
XCTAssertEqual(atom.nucleus, "c", desc);
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "{1 \\atop c}", desc);
}
func testAtopInParens() throws {
let str = "5 + {1 \\atop c} + 8";
let list = MTMathListBuilder.build(fromString:str)!
let desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc);
XCTAssertEqual((list.atoms.count), 5, desc);
let types = [MTMathAtomType.number, .binaryOperator, .fraction, .binaryOperator, .number]
self.checkAtomTypes(list, types:types, desc:desc)
let frac = list.atoms[2] as! MTFraction
XCTAssertEqual(frac.type, .fraction, desc);
XCTAssertEqual(frac.nucleus, "", desc);
XCTAssertFalse(frac.hasRule);
XCTAssertTrue(frac.rightDelimiter.isEmpty)
XCTAssertTrue(frac.leftDelimiter.isEmpty)
var subList = frac.numerator!
XCTAssertNotNil(subList, desc);
XCTAssertEqual((subList.atoms.count), 1, desc);
var atom = subList.atoms[0];
XCTAssertEqual(atom.type, .number, desc);
XCTAssertEqual(atom.nucleus, "1", desc);
atom = list.atoms[0];
subList = frac.denominator!
XCTAssertNotNil(subList, desc);
XCTAssertEqual((subList.atoms.count), 1, desc);
atom = subList.atoms[0];
XCTAssertEqual(atom.type, .variable, desc);
XCTAssertEqual(atom.nucleus, "c", desc);
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "5+{1 \\atop c}+8", desc);
}
func testChoose() throws {
let str = "n \\choose k";
let list = MTMathListBuilder.build(fromString:str)!
let desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc);
XCTAssertEqual((list.atoms.count), 1, desc);
let frac = list.atoms[0] as! MTFraction
XCTAssertEqual(frac.type, .fraction, desc);
XCTAssertEqual(frac.nucleus, "", desc);
XCTAssertFalse(frac.hasRule);
XCTAssertEqual(frac.rightDelimiter, ")");
XCTAssertEqual(frac.leftDelimiter, "(");
var subList = frac.numerator!
XCTAssertNotNil(subList, desc);
XCTAssertEqual((subList.atoms.count), 1, desc);
var atom = subList.atoms[0];
XCTAssertEqual(atom.type, .variable, desc);
XCTAssertEqual(atom.nucleus, "n", desc);
atom = list.atoms[0];
subList = frac.denominator!
XCTAssertNotNil(subList, desc);
XCTAssertEqual((subList.atoms.count), 1, desc);
atom = subList.atoms[0];
XCTAssertEqual(atom.type, .variable, desc);
XCTAssertEqual(atom.nucleus, "k", desc);
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "{n \\choose k}", desc);
}
func testBrack() throws {
let str = "n \\brack k";
let list = MTMathListBuilder.build(fromString:str)!
let desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc);
XCTAssertEqual((list.atoms.count), 1, desc);
let frac = list.atoms[0] as! MTFraction
XCTAssertEqual(frac.type, .fraction, desc);
XCTAssertEqual(frac.nucleus, "", desc);
XCTAssertFalse(frac.hasRule);
XCTAssertEqual(frac.rightDelimiter, "]");
XCTAssertEqual(frac.leftDelimiter, "[");
var subList = frac.numerator!
XCTAssertNotNil(subList, desc);
XCTAssertEqual((subList.atoms.count), 1, desc);
var atom = subList.atoms[0];
XCTAssertEqual(atom.type, .variable, desc);
XCTAssertEqual(atom.nucleus, "n", desc);
atom = list.atoms[0];
subList = frac.denominator!
XCTAssertNotNil(subList, desc);
XCTAssertEqual((subList.atoms.count), 1, desc);
atom = subList.atoms[0];
XCTAssertEqual(atom.type, .variable, desc);
XCTAssertEqual(atom.nucleus, "k", desc);
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "{n \\brack k}", desc);
}
func testBrace() throws {
let str = "n \\brace k";
let list = MTMathListBuilder.build(fromString:str)!
let desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc);
XCTAssertEqual((list.atoms.count), 1, desc);
let frac = list.atoms[0] as! MTFraction
XCTAssertEqual(frac.type, .fraction, desc);
XCTAssertEqual(frac.nucleus, "", desc);
XCTAssertFalse(frac.hasRule);
XCTAssertEqual(frac.rightDelimiter, "}");
XCTAssertEqual(frac.leftDelimiter, "{");
var subList = frac.numerator!
XCTAssertNotNil(subList, desc);
XCTAssertEqual((subList.atoms.count), 1, desc);
var atom = subList.atoms[0];
XCTAssertEqual(atom.type, .variable, desc);
XCTAssertEqual(atom.nucleus, "n", desc);
atom = list.atoms[0];
subList = frac.denominator!
XCTAssertNotNil(subList, desc);
XCTAssertEqual((subList.atoms.count), 1, desc);
atom = subList.atoms[0];
XCTAssertEqual(atom.type, .variable, desc);
XCTAssertEqual(atom.nucleus, "k", desc);
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "{n \\brace k}", desc);
}
func testBinom() throws {
let str = "\\binom{n}{k}";
let list = MTMathListBuilder.build(fromString:str)!
let desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc);
XCTAssertEqual((list.atoms.count), 1, desc);
let frac = list.atoms[0] as! MTFraction
XCTAssertEqual(frac.type, .fraction, desc);
XCTAssertEqual(frac.nucleus, "", desc);
XCTAssertFalse(frac.hasRule);
XCTAssertEqual(frac.rightDelimiter, ")");
XCTAssertEqual(frac.leftDelimiter, "(");
var subList = frac.numerator!
XCTAssertNotNil(subList, desc);
XCTAssertEqual((subList.atoms.count), 1, desc);
var atom = subList.atoms[0];
XCTAssertEqual(atom.type, .variable, desc);
XCTAssertEqual(atom.nucleus, "n", desc);
atom = list.atoms[0];
subList = frac.denominator!
XCTAssertNotNil(subList, desc);
XCTAssertEqual((subList.atoms.count), 1, desc);
atom = subList.atoms[0];
XCTAssertEqual(atom.type, .variable, desc);
XCTAssertEqual(atom.nucleus, "k", desc);
// convert it back to latex (binom converts to choose)
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "{n \\choose k}", desc);
}
func testOverLine() throws {
let str = "\\overline 2";
let list = MTMathListBuilder.build(fromString:str)!
let desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc);
XCTAssertEqual((list.atoms.count), 1, desc);
let over = list.atoms[0] as! MTOverLine
XCTAssertEqual(over.type, .overline, desc);
XCTAssertEqual(over.nucleus, "", desc);
let subList = over.innerList!
XCTAssertNotNil(subList, desc);
XCTAssertEqual((subList.atoms.count), 1, desc);
let atom = subList.atoms[0];
XCTAssertEqual(atom.type, .number, desc);
XCTAssertEqual(atom.nucleus, "2", desc);
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\overline{2}", desc);
}
func testUnderline() throws {
let str = "\\underline 2";
let list = MTMathListBuilder.build(fromString:str)!
let desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc);
XCTAssertEqual((list.atoms.count), 1, desc);
let under = list.atoms[0] as! MTUnderLine
XCTAssertEqual(under.type, .underline, desc);
XCTAssertEqual(under.nucleus, "", desc);
let subList = under.innerList!
XCTAssertNotNil(subList, desc);
XCTAssertEqual((subList.atoms.count), 1, desc);
let atom = subList.atoms[0];
XCTAssertEqual(atom.type, .number, desc);
XCTAssertEqual(atom.nucleus, "2", desc);
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\underline{2}", desc);
}
func testAccent() throws {
let str = "\\bar x";
let list = MTMathListBuilder.build(fromString:str)!
let desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc);
XCTAssertEqual((list.atoms.count), 1, desc);
let accent = list.atoms[0] as! MTAccent
XCTAssertEqual(accent.type, .accent, desc);
XCTAssertEqual(accent.nucleus, "\u{0304}", desc);
let subList = accent.innerList!
XCTAssertNotNil(subList, desc);
XCTAssertEqual((subList.atoms.count), 1, desc);
let atom = subList.atoms[0];
XCTAssertEqual(atom.type, .variable, desc);
XCTAssertEqual(atom.nucleus, "x", desc);
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\bar{x}", desc);
}
func testAccentedCharacter() throws {
let str = "á"
let list = MTMathListBuilder.build(fromString: str)!
let desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc)
XCTAssertEqual((list.atoms.count), 1, desc)
let accent = list.atoms[0] as! MTAccent
XCTAssertEqual(accent.type, .accent, desc)
XCTAssertEqual(accent.nucleus, "\u{0301}", desc)
let subList = accent.innerList!
XCTAssertNotNil(subList, desc)
XCTAssertEqual((subList.atoms.count), 1, desc)
let atom = subList.atoms[0]
XCTAssertEqual(atom.type, .variable, desc)
XCTAssertEqual(atom.nucleus, "a", desc)
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\acute{a}", desc)
}
func testMathSpace() throws {
let str = "\\!";
let list = MTMathListBuilder.build(fromString:str)!
let desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc);
XCTAssertEqual((list.atoms.count), 1, desc);
let space = list.atoms[0] as! MTMathSpace
XCTAssertEqual(space.type, .space, desc);
XCTAssertEqual(space.nucleus, "", desc);
XCTAssertEqual(space.space, -3);
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\! ", desc);
}
func testMathStyle() throws {
let str = "\\textstyle y \\scriptstyle x";
let list = MTMathListBuilder.build(fromString:str)!
let desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc);
XCTAssertEqual((list.atoms.count), 4, desc);
let style = list.atoms[0] as! MTMathStyle
XCTAssertEqual(style.type, .style, desc);
XCTAssertEqual(style.nucleus, "", desc);
XCTAssertEqual(style.style, .text);
let style2 = list.atoms[2] as! MTMathStyle
XCTAssertEqual(style2.type, .style, desc);
XCTAssertEqual(style2.nucleus, "", desc);
XCTAssertEqual(style2.style, .script);
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\textstyle y\\scriptstyle x", desc);
}
func testMatrix() throws {
let str = "\\begin{matrix} x & y \\\\ z & w \\end{matrix}";
let list = MTMathListBuilder.build(fromString:str)!
XCTAssertNotNil(list);
XCTAssertEqual((list.atoms.count), 1);
let table = list.atoms[0] as! MTMathTable
XCTAssertEqual(table.type, .table);
XCTAssertEqual(table.nucleus, "");
XCTAssertEqual(table.environment, "matrix");
XCTAssertEqual(table.interRowAdditionalSpacing, 0);
XCTAssertEqual(table.interColumnSpacing, 18);
XCTAssertEqual(table.numRows, 2);
XCTAssertEqual(table.numColumns, 2);
for i in 0..<2 {
let alignment = table.get(alignmentForColumn:i)
XCTAssertEqual(alignment, .center);
for j in 0..<2 {
let cell = table.cells[j][i];
XCTAssertEqual(cell.atoms.count, 2);
let style = cell.atoms[0] as! MTMathStyle
XCTAssertEqual(style.type, .style);
XCTAssertEqual(style.style, .text);
let atom = cell.atoms[1];
XCTAssertEqual(atom.type, .variable);
}
}
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\begin{matrix}x&y\\\\ z&w\\end{matrix}");
}
func testPMatrix() throws {
let str = "\\begin{pmatrix} x & y \\\\ z & w \\end{pmatrix}";
let list = MTMathListBuilder.build(fromString:str)!
XCTAssertNotNil(list);
XCTAssertEqual((list.atoms.count), 1);
let inner = list.atoms[0] as! MTInner
XCTAssertEqual(inner.type, .inner, str);
XCTAssertEqual(inner.nucleus, "", str);
let innerList = inner.innerList!
XCTAssertNotNil(innerList, str);
XCTAssertNotNil(inner.leftBoundary, str);
XCTAssertEqual(inner.leftBoundary!.type, .boundary, str);
XCTAssertEqual(inner.leftBoundary!.nucleus, "(", str);
XCTAssertNotNil(inner.rightBoundary, str);
XCTAssertEqual(inner.rightBoundary!.type, .boundary, str);
XCTAssertEqual(inner.rightBoundary!.nucleus, ")", str);
XCTAssertEqual((innerList.atoms.count), 1);
let table = innerList.atoms[0] as! MTMathTable
XCTAssertEqual(table.type, .table);
XCTAssertEqual(table.nucleus, "");
XCTAssertEqual(table.environment, "matrix");
XCTAssertEqual(table.interRowAdditionalSpacing, 0);
XCTAssertEqual(table.interColumnSpacing, 18);
XCTAssertEqual(table.numRows, 2);
XCTAssertEqual(table.numColumns, 2);
for i in 0..<2 {
let alignment = table.get(alignmentForColumn:i)
XCTAssertEqual(alignment, .center);
for j in 0..<2 {
let cell = table.cells[j][i];
XCTAssertEqual(cell.atoms.count, 2);
let style = cell.atoms[0] as! MTMathStyle
XCTAssertEqual(style.type, .style);
XCTAssertEqual(style.style, .text);
let atom = cell.atoms[1];
XCTAssertEqual(atom.type, .variable);
}
}
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\left( \\begin{matrix}x&y\\\\ z&w\\end{matrix}\\right) ");
}
func testDefaultTable() throws {
let str = "x \\\\ y";
let list = MTMathListBuilder.build(fromString:str)!
XCTAssertNotNil(list);
XCTAssertEqual(list.atoms.count, 1);
let table = list.atoms[0] as! MTMathTable
XCTAssertEqual(table.type, .table);
XCTAssertEqual(table.nucleus, "");
XCTAssertTrue(table.environment.isEmpty);
XCTAssertEqual(table.interRowAdditionalSpacing, 1);
XCTAssertEqual(table.interColumnSpacing, 0);
XCTAssertEqual(table.numRows, 2);
XCTAssertEqual(table.numColumns, 1);
for i in 0..<1 {
let alignment = table.get(alignmentForColumn: i)
XCTAssertEqual(alignment, .left);
for j in 0..<2 {
let cell = table.cells[j][i];
XCTAssertEqual(cell.atoms.count, 1);
let atom = cell.atoms[0];
XCTAssertEqual(atom.type, .variable);
}
}
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "x\\\\ y");
}
func testDefaultTableWithCols() throws {
let str = "x & y \\\\ z & w";
let list = MTMathListBuilder.build(fromString:str)!
XCTAssertNotNil(list);
XCTAssertEqual((list.atoms.count), 1);
let table = list.atoms[0] as! MTMathTable
XCTAssertEqual(table.type, .table);
XCTAssertEqual(table.nucleus, "");
XCTAssertTrue(table.environment.isEmpty);
XCTAssertEqual(table.interRowAdditionalSpacing, 1);
XCTAssertEqual(table.interColumnSpacing, 0);
XCTAssertEqual(table.numRows, 2);
XCTAssertEqual(table.numColumns, 2);
for i in 0..<2 {
let alignment = table.get(alignmentForColumn:i)
XCTAssertEqual(alignment, .left);
for j in 0..<2 {
let cell = table.cells[j][i];
XCTAssertEqual(cell.atoms.count, 1);
let atom = cell.atoms[0];
XCTAssertEqual(atom.type, .variable);
}
}
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "x&y\\\\ z&w");
}
func testEqalign() throws {
let str1 = "\\begin{eqalign}x&y\\\\ z&w\\end{eqalign}";
let str2 = "\\begin{split}x&y\\\\ z&w\\end{split}";
let str3 = "\\begin{aligned}x&y\\\\ z&w\\end{aligned}";
for str in [str1, str2, str3] {
let list = MTMathListBuilder.build(fromString:str)!
XCTAssertNotNil(list);
XCTAssertEqual((list.atoms.count), 1);
let table = list.atoms[0] as! MTMathTable
XCTAssertEqual(table.type, .table);
XCTAssertEqual(table.nucleus, "");
XCTAssertEqual(table.interRowAdditionalSpacing, 1);
XCTAssertEqual(table.interColumnSpacing, 0);
XCTAssertEqual(table.numRows, 2);
XCTAssertEqual(table.numColumns, 2);
for i in 0..<2 {
let alignment = table.get(alignmentForColumn:i)
XCTAssertEqual(alignment, (i == 0) ? .right: .left);
for j in 0..<2 {
let cell = table.cells[j][i];
if (i == 0) {
XCTAssertEqual(cell.atoms.count, 1);
let atom = cell.atoms[0];
XCTAssertEqual(atom.type, .variable);
} else {
XCTAssertEqual(cell.atoms.count, 2);
self.checkAtomTypes(cell, types:[.ordinary, .variable], desc:str)
}
}
}
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, str);
}
}
func testDisplayLines() throws {
let str1 = "\\begin{displaylines}x\\\\ y\\end{displaylines}";
let str2 = "\\begin{gather}x\\\\ y\\end{gather}";
for str in [str1, str2] {
let list = MTMathListBuilder.build(fromString:str)
XCTAssertNotNil(list)
XCTAssertEqual(list?.atoms.count, 1);
let table = list?.atoms[0] as! MTMathTable
XCTAssertEqual(table.type, .table);
XCTAssertEqual(table.nucleus, "");
XCTAssertEqual(table.interRowAdditionalSpacing, 1);
XCTAssertEqual(table.interColumnSpacing, 0);
XCTAssertEqual(table.numRows, 2);
XCTAssertEqual(table.numColumns, 1);
for i in 0..<1 {
let alignment = table.get(alignmentForColumn:i)
XCTAssertEqual(alignment, .center);
for j in 0..<2 {
let cell = table.cells[j][i];
XCTAssertEqual(cell.atoms.count, 1);
let atom = cell.atoms[0];
XCTAssertEqual(atom.type, .variable);
}
}
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, str);
}
}
func testErrors() throws {
let data = getTestDataParseErrors()
for testCase in data {
let str = testCase.0
var error : NSError? = nil
let list = MTMathListBuilder.build(fromString: str, error:&error)
let desc = "Error for string:\(str)"
XCTAssertNil(list, desc)
XCTAssertNotNil(error, desc)
XCTAssertEqual(error!.domain, MTParseError, desc)
let num = testCase.1
XCTAssertEqual(error!.code, num.rawValue, desc)
}
}
func testCustom() throws {
let str = "\\lcm(a,b)";
var error : NSError? = nil
var list = MTMathListBuilder.build(fromString: str, error:&error)
XCTAssertNil(list)
XCTAssertNotNil(error)
MTMathAtomFactory.add(latexSymbol: "lcm", value: MTMathAtomFactory.operatorWithName("lcm", limits:false))
error = nil
list = MTMathListBuilder.build(fromString: str, error:&error)
let atomTypes = [MTMathAtomType.largeOperator, .open, .variable, .punctuation, .variable, .close]
self.checkAtomTypes(list, types:atomTypes, desc:"Error for lcm")
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\lcm (a,b)");
}
func testFontSingle() throws {
let str = "\\mathbf x";
let list = MTMathListBuilder.build(fromString: str)!
let desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc)
XCTAssertEqual(list.atoms.count, 1, desc)
let atom = list.atoms[0];
XCTAssertEqual(atom.type, .variable, desc)
XCTAssertEqual(atom.nucleus, "x", desc)
XCTAssertEqual(atom.fontStyle, .bold)
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\mathbf{x}", desc)
}
func testFontOneChar() throws {
let str = "\\cal xy";
let list = MTMathListBuilder.build(fromString: str)!
let desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc)
XCTAssertEqual((list.atoms.count), 2, desc)
var atom = list.atoms[0];
XCTAssertEqual(atom.type, .variable, desc)
XCTAssertEqual(atom.nucleus, "x", desc)
XCTAssertEqual(atom.fontStyle, .caligraphic);
atom = list.atoms[1];
XCTAssertEqual(atom.type, .variable, desc)
XCTAssertEqual(atom.nucleus, "y", desc)
XCTAssertEqual(atom.fontStyle, .defaultStyle);
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\mathcal{x}y", desc)
}
func testFontMultipleChars() throws {
let str = "\\frak{xy}";
let list = MTMathListBuilder.build(fromString: str)!
let desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc)
XCTAssertEqual((list.atoms.count), 2, desc)
var atom = list.atoms[0];
XCTAssertEqual(atom.type, .variable, desc)
XCTAssertEqual(atom.nucleus, "x", desc)
XCTAssertEqual(atom.fontStyle, .fraktur);
atom = list.atoms[1];
XCTAssertEqual(atom.type, .variable, desc)
XCTAssertEqual(atom.nucleus, "y", desc)
XCTAssertEqual(atom.fontStyle, .fraktur);
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\mathfrak{xy}", desc)
}
func testFontOneCharInside() throws {
let str = "\\sqrt \\mathrm x y";
let list = MTMathListBuilder.build(fromString: str)!
let desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc)
XCTAssertEqual((list.atoms.count), 2, desc)
let rad = list.atoms[0] as! MTRadical
XCTAssertEqual(rad.type, .radical, desc)
XCTAssertEqual(rad.nucleus, "", desc)
let subList = rad.radicand!
var atom = subList.atoms[0];
XCTAssertEqual(atom.type, .variable, desc)
XCTAssertEqual(atom.nucleus, "x", desc)
XCTAssertEqual(atom.fontStyle, .roman);
atom = list.atoms[1];
XCTAssertEqual(atom.type, .variable, desc)
XCTAssertEqual(atom.nucleus, "y", desc)
XCTAssertEqual(atom.fontStyle, .defaultStyle)
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\sqrt{\\mathrm{x}}y", desc)
}
func testText() throws {
let str = "\\text{x y}";
let list = MTMathListBuilder.build(fromString: str)!
let desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc)
XCTAssertEqual((list.atoms.count), 3, desc)
var atom = list.atoms[0];
XCTAssertEqual(atom.type, .variable, desc)
XCTAssertEqual(atom.nucleus, "x", desc)
XCTAssertEqual(atom.fontStyle, .roman);
atom = list.atoms[1];
XCTAssertEqual(atom.type, .ordinary, desc)
XCTAssertEqual(atom.nucleus, " ", desc)
atom = list.atoms[2];
XCTAssertEqual(atom.type, .variable, desc)
XCTAssertEqual(atom.nucleus, "y", desc)
XCTAssertEqual(atom.fontStyle, .roman);
// convert it back to latex
let latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\mathrm{x\\ y}", desc)
}
func testLimits() throws {
// Int with no limits (default)
var str = "\\int";
var list = MTMathListBuilder.build(fromString: str)!
var desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc)
XCTAssertEqual((list.atoms.count), 1, desc)
var op = list.atoms[0] as! MTLargeOperator
XCTAssertEqual(op.type, .largeOperator, desc)
XCTAssertFalse(op.limits);
// convert it back to latex
var latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\int ", desc)
// Int with limits
str = "\\int\\limits"
list = MTMathListBuilder.build(fromString: str)!
desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc)
XCTAssertEqual((list.atoms.count), 1, desc)
op = list.atoms[0] as! MTLargeOperator
XCTAssertEqual(op.type, .largeOperator, desc)
XCTAssertTrue(op.limits)
// convert it back to latex
latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\int \\limits ", desc)
}
func testNoLimits() throws {
// Sum with limits (default)
var str = "\\sum";
var list = MTMathListBuilder.build(fromString: str)!
var desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc)
XCTAssertEqual((list.atoms.count), 1, desc)
var op = list.atoms[0] as! MTLargeOperator
XCTAssertEqual(op.type, .largeOperator, desc)
XCTAssertTrue(op.limits);
// convert it back to latex
var latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\sum ", desc)
// Int with limits
str = "\\sum\\nolimits";
list = MTMathListBuilder.build(fromString: str)!
desc = "Error for string:\(str)"
XCTAssertNotNil(list, desc)
XCTAssertEqual(list.atoms.count, 1, desc)
op = list.atoms[0] as! MTLargeOperator
XCTAssertEqual(op.type, .largeOperator, desc)
XCTAssertFalse(op.limits);
// convert it back to latex
latex = MTMathListBuilder.mathListToString(list)
XCTAssertEqual(latex, "\\sum \\nolimits ", desc)
}
// MARK: - Inline and Display Math Delimiter Tests
func testInlineMathDollar() throws {
let str = "$x^2$"
let list = MTMathListBuilder.build(fromString: str)
XCTAssertNotNil(list, "Should parse inline math with $")
// Should have textstyle at start, then variable with superscript
XCTAssertTrue(list!.atoms.count >= 1, "Should have at least one atom")
// Find the variable atom (skip style atoms)
var foundVariable = false
for atom in list!.atoms {
if atom.type == .variable && atom.nucleus == "x" {
foundVariable = true
XCTAssertNotNil(atom.superScript, "Should have superscript")
break
}
}
XCTAssertTrue(foundVariable, "Should find variable x")
}
func testInlineMathParens() throws {
let str = "\\(E=mc^2\\)"
let list = MTMathListBuilder.build(fromString: str)
XCTAssertNotNil(list, "Should parse inline math with \\(\\)")
XCTAssertTrue(list!.atoms.count >= 3, "Should have E, =, m, c atoms")
// Check for equals sign
var foundEquals = false
for atom in list!.atoms {
if atom.type == .relation && atom.nucleus == "=" {
foundEquals = true
break
}
}
XCTAssertTrue(foundEquals, "Should find equals sign")
}
func testInlineMathWithCases() throws {
let str = "\\(\\begin{cases} x + y = 5 \\\\ 2x - y = 1 \\end{cases}\\)"
let list = MTMathListBuilder.build(fromString: str)
XCTAssertNotNil(list, "Should parse inline cases")
// cases environment returns an Inner atom with table inside
var foundInner = false
for atom in list!.atoms {
if atom.type == .inner {
let inner = atom as! MTInner
// Look for table inside the inner list
if let innerList = inner.innerList {
for innerAtom in innerList.atoms {
if innerAtom.type == .table {
let table = innerAtom as! MTMathTable
XCTAssertEqual(table.environment, "cases", "Should be cases environment")
XCTAssertEqual(table.numRows, 2, "Should have 2 rows")
foundInner = true
break
}
}
}
if foundInner { break }
}
}
XCTAssertTrue(foundInner, "Should find cases table inside inner atom")
}
func testInlineMathVectorDot() throws {
let str = "$\\vec{a} \\cdot \\vec{b}$"
let list = MTMathListBuilder.build(fromString: str)
XCTAssertNotNil(list, "Should parse inline vector dot product")
// Should contain accents (for vec) and cdot operator
var hasAccent = false
var hasCdot = false
for atom in list!.atoms {
if atom.type == .accent {
hasAccent = true
}
if atom.type == .binaryOperator && atom.nucleus.contains("\u{22C5}") {
hasCdot = true
}
}
XCTAssertTrue(hasAccent, "Should have accent for \\vec")
XCTAssertTrue(hasCdot, "Should have \\cdot operator")
}
func testDisplayMathDoubleDollar() throws {
let str = "$$x^2 + y^2 = z^2$$"
let list = MTMathListBuilder.build(fromString: str)
XCTAssertNotNil(list, "Should parse display math with $$")
XCTAssertTrue(list!.atoms.count >= 5, "Should have multiple atoms for expression")
// Should NOT have textstyle at start (display mode)
let firstAtom = list!.atoms.first
XCTAssertNotEqual(firstAtom?.type, .style, "Display mode should not force textstyle")
}
func testDisplayMathBrackets() throws {
let str = "\\[\\sum_{i=1}^{n} i = \\frac{n(n+1)}{2}\\]"
let list = MTMathListBuilder.build(fromString: str)
XCTAssertNotNil(list, "Should parse display math with \\[\\]")
// Find sum operator
var foundSum = false
for atom in list!.atoms {
if atom.type == .largeOperator && atom.nucleus.contains("") {
foundSum = true
XCTAssertNotNil(atom.subScript, "Sum should have subscript")
XCTAssertNotNil(atom.superScript, "Sum should have superscript")
break
}
}
XCTAssertTrue(foundSum, "Should find sum operator")
}
func testDisplayMathCasesWithoutDelimiters() throws {
// This should work as before (backward compatibility)
let str = "\\begin{cases} x + y = 5 \\\\ 2x - y = 1 \\end{cases}"
let list = MTMathListBuilder.build(fromString: str)
XCTAssertNotNil(list, "Should parse display cases without outer delimiters")
XCTAssertTrue(list!.atoms.count >= 1, "Should have at least one atom")
// cases environment returns an Inner atom with table inside
var foundTable = false
for atom in list!.atoms {
if atom.type == .inner {
let inner = atom as! MTInner
if let innerList = inner.innerList {
for innerAtom in innerList.atoms {
if innerAtom.type == .table {
let table = innerAtom as! MTMathTable
XCTAssertEqual(table.environment, "cases", "Should be cases environment")
XCTAssertEqual(table.numRows, 2, "Should have 2 rows")
foundTable = true
break
}
}
}
if foundTable { break }
}
}
XCTAssertTrue(foundTable, "Should find cases table inside inner atom")
}
func testBackwardCompatibilityNoDelimiters() throws {
// Test that expressions without delimiters still work
let str = "x^2 + y^2 = z^2"
let list = MTMathListBuilder.build(fromString: str)
XCTAssertNotNil(list, "Should parse expression without delimiters")
XCTAssertTrue(list!.atoms.count >= 5, "Should have multiple atoms")
}
func testEmptyInlineMath() throws {
let str = "$$$" // This is $$$ which should be treated as $$ + $
let list = MTMathListBuilder.build(fromString: str)
// Should handle gracefully
XCTAssertNotNil(list, "Should handle edge case")
}
func testEmptyDisplayMath() throws {
let str = "\\[\\]"
let list = MTMathListBuilder.build(fromString: str)
// Empty content may return nil or an empty list, both are acceptable
if list != nil {
XCTAssertTrue(list!.atoms.isEmpty || list!.atoms.count >= 0, "Should have empty or minimal atoms")
}
// It's ok if it returns nil for empty content
}
func testDollarInMath() throws {
// Test that delimiters are properly stripped
let str = "$a + b$"
let list = MTMathListBuilder.build(fromString: str)
XCTAssertNotNil(list, "Should parse correctly")
// Should not contain $ in the parsed atoms
for atom in list!.atoms {
XCTAssertFalse(atom.nucleus.contains("$"), "Should not have $ in nucleus")
}
}
func testComplexInlineExpression() throws {
let str = "$\\frac{1}{2} + \\sqrt{3}$"
let list = MTMathListBuilder.build(fromString: str)
XCTAssertNotNil(list, "Should parse complex inline expression")
// Should have fraction and radical
var hasFraction = false
var hasRadical = false
for atom in list!.atoms {
if atom.type == .fraction {
hasFraction = true
}
if atom.type == .radical {
hasRadical = true
}
}
XCTAssertTrue(hasFraction, "Should have fraction")
XCTAssertTrue(hasRadical, "Should have radical")
}
func testInlineMathStyleForcing() throws {
// Inline math should have textstyle prepended
let str = "$\\sum_{i=1}^{n} i$"
let list = MTMathListBuilder.build(fromString: str)
XCTAssertNotNil(list, "Should parse sum in inline mode")
// First atom should be style atom with text style
if let firstAtom = list!.atoms.first, firstAtom.type == .style {
let styleAtom = firstAtom as! MTMathStyle
XCTAssertEqual(styleAtom.style, .text, "Inline mode should force text style")
}
}
// MARK: - Tests for build(fromString:error:) API with delimiters
func testInlineMathDollarWithError() throws {
let str = "$x^2$"
var error: NSError? = nil
let list = MTMathListBuilder.build(fromString: str, error: &error)
XCTAssertNotNil(list, "Should parse inline math with $")
XCTAssertNil(error, "Should not have error")
// Find the variable atom (skip style atoms)
var foundVariable = false
for atom in list!.atoms {
if atom.type == .variable && atom.nucleus == "x" {
foundVariable = true
XCTAssertNotNil(atom.superScript, "Should have superscript")
break
}
}
XCTAssertTrue(foundVariable, "Should find variable x")
}
func testInlineMathParensWithError() throws {
let str = "\\(E=mc^2\\)"
var error: NSError? = nil
let list = MTMathListBuilder.build(fromString: str, error: &error)
XCTAssertNotNil(list, "Should parse inline math with \\(\\)")
XCTAssertNil(error, "Should not have error")
XCTAssertTrue(list!.atoms.count >= 3, "Should have E, =, m, c atoms")
// Check for equals sign
var foundEquals = false
for atom in list!.atoms {
if atom.type == .relation && atom.nucleus == "=" {
foundEquals = true
break
}
}
XCTAssertTrue(foundEquals, "Should find equals sign")
}
func testInlineMathWithCasesWithError() throws {
let str = "\\(\\begin{cases} x + y = 5 \\\\ 2x - y = 1 \\end{cases}\\)"
var error: NSError? = nil
let list = MTMathListBuilder.build(fromString: str, error: &error)
XCTAssertNotNil(list, "Should parse inline cases")
XCTAssertNil(error, "Should not have error")
// cases environment returns an Inner atom with table inside
var foundInner = false
for atom in list!.atoms {
if atom.type == .inner {
let inner = atom as! MTInner
if let innerList = inner.innerList {
for innerAtom in innerList.atoms {
if innerAtom.type == .table {
let table = innerAtom as! MTMathTable
XCTAssertEqual(table.environment, "cases", "Should be cases environment")
XCTAssertEqual(table.numRows, 2, "Should have 2 rows")
foundInner = true
break
}
}
}
if foundInner { break }
}
}
XCTAssertTrue(foundInner, "Should find cases table inside inner atom")
}
func testDisplayMathDoubleDollarWithError() throws {
let str = "$$x^2 + y^2 = z^2$$"
var error: NSError? = nil
let list = MTMathListBuilder.build(fromString: str, error: &error)
XCTAssertNotNil(list, "Should parse display math with $$")
XCTAssertNil(error, "Should not have error")
XCTAssertTrue(list!.atoms.count >= 5, "Should have multiple atoms for expression")
}
func testDisplayMathBracketsWithError() throws {
let str = "\\[\\sum_{i=1}^{n} i = \\frac{n(n+1)}{2}\\]"
var error: NSError? = nil
let list = MTMathListBuilder.build(fromString: str, error: &error)
XCTAssertNotNil(list, "Should parse display math with \\[\\]")
XCTAssertNil(error, "Should not have error")
// Find sum operator
var foundSum = false
for atom in list!.atoms {
if atom.type == .largeOperator && atom.nucleus.contains("") {
foundSum = true
XCTAssertNotNil(atom.subScript, "Sum should have subscript")
XCTAssertNotNil(atom.superScript, "Sum should have superscript")
break
}
}
XCTAssertTrue(foundSum, "Should find sum operator")
}
func testDisplayMathCasesWithoutDelimitersWithError() throws {
let str = "\\begin{cases} x + y = 5 \\\\ 2x - y = 1 \\end{cases}"
var error: NSError? = nil
let list = MTMathListBuilder.build(fromString: str, error: &error)
XCTAssertNotNil(list, "Should parse display cases without outer delimiters")
XCTAssertNil(error, "Should not have error")
XCTAssertTrue(list!.atoms.count >= 1, "Should have at least one atom")
// cases environment returns an Inner atom with table inside
var foundTable = false
for atom in list!.atoms {
if atom.type == .inner {
let inner = atom as! MTInner
if let innerList = inner.innerList {
for innerAtom in innerList.atoms {
if innerAtom.type == .table {
let table = innerAtom as! MTMathTable
XCTAssertEqual(table.environment, "cases", "Should be cases environment")
XCTAssertEqual(table.numRows, 2, "Should have 2 rows")
foundTable = true
break
}
}
}
if foundTable { break }
}
}
XCTAssertTrue(foundTable, "Should find cases table inside inner atom")
}
func testBackwardCompatibilityNoDelimitersWithError() throws {
let str = "x^2 + y^2 = z^2"
var error: NSError? = nil
let list = MTMathListBuilder.build(fromString: str, error: &error)
XCTAssertNotNil(list, "Should parse expression without delimiters")
XCTAssertNil(error, "Should not have error")
XCTAssertTrue(list!.atoms.count >= 5, "Should have multiple atoms")
}
func testInvalidLatexWithError() throws {
let str = "$\\notacommand$"
var error: NSError? = nil
let list = MTMathListBuilder.build(fromString: str, error: &error)
XCTAssertNil(list, "Should fail to parse invalid command")
XCTAssertNotNil(error, "Should have error")
XCTAssertEqual(error?.code, MTParseErrors.invalidCommand.rawValue, "Should be invalid command error")
}
func testMismatchedBracesWithError() throws {
let str = "${x+2$"
var error: NSError? = nil
let list = MTMathListBuilder.build(fromString: str, error: &error)
XCTAssertNil(list, "Should fail to parse mismatched braces")
XCTAssertNotNil(error, "Should have error")
XCTAssertEqual(error?.code, MTParseErrors.mismatchBraces.rawValue, "Should be mismatched braces error")
}
func testComplexInlineExpressionWithError() throws {
let str = "$\\frac{1}{2} + \\sqrt{3}$"
var error: NSError? = nil
let list = MTMathListBuilder.build(fromString: str, error: &error)
XCTAssertNotNil(list, "Should parse complex inline expression")
XCTAssertNil(error, "Should not have error")
// Should have fraction and radical
var hasFraction = false
var hasRadical = false
for atom in list!.atoms {
if atom.type == .fraction {
hasFraction = true
}
if atom.type == .radical {
hasRadical = true
}
}
XCTAssertTrue(hasFraction, "Should have fraction")
XCTAssertTrue(hasRadical, "Should have radical")
}
func testInlineMathVectorDotWithError() throws {
let str = "$\\vec{a} \\cdot \\vec{b}$"
var error: NSError? = nil
let list = MTMathListBuilder.build(fromString: str, error: &error)
XCTAssertNotNil(list, "Should parse inline vector dot product")
XCTAssertNil(error, "Should not have error")
// Should contain accents (for vec) and cdot operator
var hasAccent = false
var hasCdot = false
for atom in list!.atoms {
if atom.type == .accent {
hasAccent = true
}
if atom.type == .binaryOperator && atom.nucleus.contains("\u{22C5}") {
hasCdot = true
}
}
XCTAssertTrue(hasAccent, "Should have accent for \\vec")
XCTAssertTrue(hasCdot, "Should have \\cdot operator")
}
// MARK: - Comprehensive Command Coverage Tests
func testGreekLettersLowercase() throws {
let commands = ["alpha", "beta", "gamma", "delta", "epsilon", "zeta", "eta", "theta",
"iota", "kappa", "lambda", "mu", "nu", "xi", "omicron", "pi",
"rho", "sigma", "tau", "upsilon", "phi", "chi", "psi", "omega"]
for cmd in commands {
var error: NSError? = nil
let str = "$\\\(cmd)$"
let list = MTMathListBuilder.build(fromString: str, error: &error)
let unwrappedList = try XCTUnwrap(list, "Should parse \\\(cmd)")
XCTAssertNil(error, "Should not error on \\\(cmd): \(error?.localizedDescription ?? "")")
XCTAssertTrue(unwrappedList.atoms.count >= 1, "\\\(cmd) should have at least one atom")
}
}
func testGreekLettersUppercase() throws {
let commands = ["Gamma", "Delta", "Theta", "Lambda", "Xi", "Pi", "Sigma", "Upsilon", "Phi", "Psi", "Omega"]
for cmd in commands {
var error: NSError? = nil
let str = "$\\\(cmd)$"
let list = MTMathListBuilder.build(fromString: str, error: &error)
let unwrappedList = try XCTUnwrap(list, "Should parse \\\(cmd)")
XCTAssertNil(error, "Should not error on \\\(cmd): \(error?.localizedDescription ?? "")")
XCTAssertTrue(unwrappedList.atoms.count >= 1, "\\\(cmd) should have at least one atom")
}
}
func testBinaryOperators() throws {
let operators = ["times", "div", "pm", "mp", "ast", "star", "circ", "bullet",
"cdot", "cap", "cup", "uplus", "sqcap", "sqcup",
"oplus", "ominus", "otimes", "oslash", "odot", "wedge", "vee"]
for op in operators {
var error: NSError? = nil
let str = "$a \\\(op) b$"
let list = MTMathListBuilder.build(fromString: str, error: &error)
let unwrappedList = try XCTUnwrap(list, "Should parse \\\(op)")
XCTAssertNil(error, "Should not error on \\\(op): \(error?.localizedDescription ?? "")")
// Should find the operator
var foundOp = false
for atom in unwrappedList.atoms {
if atom.type == .binaryOperator {
foundOp = true
break
}
}
XCTAssertTrue(foundOp, "Should find binary operator for \\\(op)")
}
}
func testRelations() throws {
let relations = ["leq", "geq", "neq", "equiv", "approx", "sim", "simeq", "cong",
"prec", "succ", "subset", "supset", "subseteq", "supseteq",
"in", "notin", "ni", "propto", "perp", "parallel"]
for rel in relations {
var error: NSError? = nil
let str = "$a \\\(rel) b$"
let list = MTMathListBuilder.build(fromString: str, error: &error)
let unwrappedList = try XCTUnwrap(list, "Should parse \\\(rel)")
XCTAssertNil(error, "Should not error on \\\(rel): \(error?.localizedDescription ?? "")")
// Should find the relation
var foundRel = false
for atom in unwrappedList.atoms {
if atom.type == .relation {
foundRel = true
break
}
}
XCTAssertTrue(foundRel, "Should find relation for \\\(rel)")
}
}
func testAllAccents() throws {
let accents = ["hat", "tilde", "bar", "dot", "ddot", "check", "grave", "acute", "breve", "vec"]
for acc in accents {
var error: NSError? = nil
let str = "$\\\(acc){x}$"
let list = MTMathListBuilder.build(fromString: str, error: &error)
let unwrappedList = try XCTUnwrap(list, "Should parse \\\(acc)")
XCTAssertNil(error, "Should not error on \\\(acc): \(error?.localizedDescription ?? "")")
// Should find the accent
var foundAccent = false
for atom in unwrappedList.atoms {
if atom.type == .accent {
foundAccent = true
break
}
}
XCTAssertTrue(foundAccent, "Should find accent for \\\(acc)")
}
}
func testDelimiterPairs() throws {
let delimiterPairs = [
("langle", "rangle"),
("lfloor", "rfloor"),
("lceil", "rceil"),
("lgroup", "rgroup"),
("{", "}")
]
for (left, right) in delimiterPairs {
var error: NSError? = nil
let str = "$\\left\\\(left) x \\right\\\(right)$"
let list = MTMathListBuilder.build(fromString: str, error: &error)
let unwrappedList = try XCTUnwrap(list, "Should parse \\left\\\(left) ... \\right\\\(right)")
XCTAssertNil(error, "Should not error on delimiters \\\(left)/\\\(right): \(error?.localizedDescription ?? "")")
// Should have an inner atom
var foundInner = false
for atom in unwrappedList.atoms {
if atom.type == .inner {
foundInner = true
break
}
}
XCTAssertTrue(foundInner, "Should create inner atom for \\left\\\(left)...\\right\\\(right)")
}
}
func testLargeOperators() throws {
let operators = ["sum", "prod", "coprod", "int", "iint", "iiint", "oint",
"bigcap", "bigcup", "bigvee", "bigwedge", "bigodot", "bigoplus", "bigotimes"]
for op in operators {
var error: NSError? = nil
let str = "$\\\(op)_{i=1}^{n} x_i$"
let list = MTMathListBuilder.build(fromString: str, error: &error)
let unwrappedList = try XCTUnwrap(list, "Should parse \\\(op)")
XCTAssertNil(error, "Should not error on \\\(op): \(error?.localizedDescription ?? "")")
// Should find large operator
var foundOp = false
for atom in unwrappedList.atoms {
if atom.type == .largeOperator {
foundOp = true
break
}
}
XCTAssertTrue(foundOp, "Should find large operator for \\\(op)")
}
}
func testArrows() throws {
let arrows = ["leftarrow", "rightarrow", "uparrow", "downarrow", "leftrightarrow",
"Leftarrow", "Rightarrow", "Uparrow", "Downarrow", "Leftrightarrow",
"longleftarrow", "longrightarrow", "Longleftarrow", "Longrightarrow",
"mapsto", "nearrow", "searrow", "swarrow", "nwarrow"]
for arrow in arrows {
var error: NSError? = nil
let str = "$a \\\(arrow) b$"
let list = MTMathListBuilder.build(fromString: str, error: &error)
let unwrappedList = try XCTUnwrap(list, "Should parse \\\(arrow)")
XCTAssertNil(error, "Should not error on \\\(arrow): \(error?.localizedDescription ?? "")")
// Arrows are typically relations
var foundArrow = false
for atom in unwrappedList.atoms {
if atom.type == .relation {
foundArrow = true
break
}
}
XCTAssertTrue(foundArrow, "Should find arrow relation for \\\(arrow)")
}
}
func testTrigonometricFunctions() throws {
let functions = ["sin", "cos", "tan", "cot", "sec", "csc",
"arcsin", "arccos", "arctan", "sinh", "cosh", "tanh", "coth"]
for funcName in functions {
var error: NSError? = nil
let str = "$\\\(funcName) x$"
let list = MTMathListBuilder.build(fromString: str, error: &error)
let unwrappedList = try XCTUnwrap(list, "Should parse \\\(funcName)")
XCTAssertNil(error, "Should not error on \\\(funcName): \(error?.localizedDescription ?? "")")
// Should find the function operator
var foundFunc = false
for atom in unwrappedList.atoms {
if atom.type == .largeOperator {
foundFunc = true
break
}
}
XCTAssertTrue(foundFunc, "Should find function operator for \\\(funcName)")
}
}
func testLimitOperators() throws {
let operators = ["lim", "limsup", "liminf", "max", "min", "sup", "inf", "det", "gcd"]
for op in operators {
var error: NSError? = nil
let str = "$\\\(op)_{x \\to 0} f(x)$"
let list = MTMathListBuilder.build(fromString: str, error: &error)
let unwrappedList = try XCTUnwrap(list, "Should parse \\\(op)")
XCTAssertNil(error, "Should not error on \\\(op): \(error?.localizedDescription ?? "")")
// Should find the operator
var foundOp = false
for atom in unwrappedList.atoms {
if atom.type == .largeOperator {
foundOp = true
break
}
}
XCTAssertTrue(foundOp, "Should find limit operator for \\\(op)")
}
}
func testSpecialSymbols() throws {
let symbols = ["infty", "partial", "nabla", "prime", "hbar", "ell", "wp",
"Re", "Im", "top", "bot", "emptyset", "exists", "forall",
"neg", "angle", "triangle", "ldots", "cdots", "vdots", "ddots"]
for sym in symbols {
var error: NSError? = nil
let str = "$\\\(sym)$"
let list = MTMathListBuilder.build(fromString: str, error: &error)
let unwrappedList = try XCTUnwrap(list, "Should parse \\\(sym)")
XCTAssertNil(error, "Should not error on \\\(sym): \(error?.localizedDescription ?? "")")
XCTAssertTrue(unwrappedList.atoms.count >= 1, "\\\(sym) should have at least one atom")
}
}
func testLogFunctions() throws {
let logFuncs = ["log", "ln", "lg"]
for funcName in logFuncs {
var error: NSError? = nil
let str = "$\\\(funcName) x$"
let list = MTMathListBuilder.build(fromString: str, error: &error)
XCTAssertNotNil(list, "Should parse \\\(funcName)")
XCTAssertNil(error, "Should not error on \\\(funcName): \(error?.localizedDescription ?? "")")
}
}
// MARK: - High Priority Missing Features Tests
func testDisplayStyle() throws {
// Test \displaystyle and \textstyle commands
let testCases = [
("\\displaystyle \\sum_{i=1}^{n} x_i", "displaystyle with sum"),
("\\textstyle \\int_{0}^{\\infty} f(x) dx", "textstyle with integral"),
("x + \\displaystyle\\frac{a}{b} + y", "inline displaystyle fraction"),
("\\displaystyle x = \\frac{-b \\pm \\sqrt{b^2-4ac}}{2a}", "displaystyle equation")
]
for (latex, desc) in testCases {
var error: NSError? = nil
let list = MTMathListBuilder.build(fromString: latex, error: &error)
if list == nil || error != nil {
throw XCTSkip("\\displaystyle/\\textstyle not implemented: \(desc). Error: \(error?.localizedDescription ?? "nil result")")
}
let unwrappedList = try XCTUnwrap(list, "Should parse: \(desc)")
XCTAssertTrue(unwrappedList.atoms.count >= 1, "\(desc) should have atoms")
}
}
func testMiddleDelimiter() throws {
// Test \middle command for delimiters in the middle of expressions
let testCases = [
("\\left( \\frac{a}{b} \\middle| \\frac{c}{d} \\right)", "middle pipe"),
("\\left\\{ x \\middle\\| y \\right\\}", "middle double pipe"),
("\\left[ a \\middle\\\\ b \\right]", "middle backslash")
]
for (latex, desc) in testCases {
var error: NSError? = nil
let list = MTMathListBuilder.build(fromString: latex, error: &error)
if list == nil || error != nil {
throw XCTSkip("\\middle not implemented: \(desc). Error: \(error?.localizedDescription ?? "nil result")")
}
let unwrappedList = try XCTUnwrap(list, "Should parse: \(desc)")
XCTAssertTrue(unwrappedList.atoms.count >= 1, "\(desc) should have atoms")
}
}
func testSubstack() throws {
// Test \substack for multi-line subscripts and limits
let testCases = [
("\\substack{a \\\\ b}", "simple substack"),
("x_{\\substack{a \\\\ b}}", "substack in subscript"),
("\\sum_{\\substack{0 \\le i \\le m \\\\ 0 < j < n}} P(i,j)", "substack in sum limits"),
("\\prod_{\\substack{p \\text{ prime} \\\\ p < 100}} p", "substack with text"),
("A_{\\substack{n \\\\ k}}", "subscript with substack"),
("\\substack{\\frac{a}{b} \\\\ c}", "substack with frac"),
("\\substack{a}", "single row substack"),
("\\substack{a \\\\ b \\\\ c \\\\ d}", "multi-row substack")
]
for (latex, desc) in testCases {
var error: NSError? = nil
let list = MTMathListBuilder.build(fromString: latex, error: &error)
let unwrappedList = try XCTUnwrap(list, "Should parse: \(desc)")
XCTAssertNil(error, "Should not error on \(desc): \(error?.localizedDescription ?? "")")
XCTAssertTrue(unwrappedList.atoms.count >= 1, "\(desc) should have atoms")
// Verify we have a table structure (either directly or in subscript)
var foundTable = false
for atom in unwrappedList.atoms {
if atom.type == .table {
foundTable = true
break
}
if let subScript = atom.subScript {
for subAtom in subScript.atoms {
if subAtom.type == .table {
foundTable = true
break
}
}
}
}
XCTAssertTrue(foundTable, "\(desc) should contain a table structure")
}
}
func testManualDelimiterSizing() throws {
// Test \big, \Big, \bigg, \Bigg sizing commands
let testCases = [
("\\big( x \\big)", "big parentheses"),
("\\Big[ y \\Big]", "Big brackets"),
("\\bigg\\{ z \\bigg\\}", "bigg braces"),
("\\Bigg| w \\Bigg|", "Bigg pipes"),
("\\big< a \\big>", "big angle brackets")
]
for (latex, desc) in testCases {
var error: NSError? = nil
let list = MTMathListBuilder.build(fromString: latex, error: &error)
if list == nil || error != nil {
throw XCTSkip("Manual delimiter sizing (\\big, \\Big, \\bigg, \\Bigg) not implemented: \(desc). Error: \(error?.localizedDescription ?? "nil result")")
}
let unwrappedList = try XCTUnwrap(list, "Should parse: \(desc)")
XCTAssertTrue(unwrappedList.atoms.count >= 1, "\(desc) should have atoms")
}
}
func testSpacingCommands() throws {
// Test fine-tuned spacing commands
let testCases = [
("a\\,b", "thin space \\,"),
("a\\:b", "medium space \\:"),
("a\\;b", "thick space \\;"),
("a\\!b", "negative space \\!"),
("\\int\\!\\!\\!\\int f(x,y) dx dy", "multiple negative spaces"),
("x \\, y \\: z \\; w", "mixed spacing")
]
for (latex, desc) in testCases {
var error: NSError? = nil
let list = MTMathListBuilder.build(fromString: latex, error: &error)
if list == nil || error != nil {
throw XCTSkip("Spacing commands (\\,, \\:, \\;, \\!) not implemented: \(desc). Error: \(error?.localizedDescription ?? "nil result")")
}
let unwrappedList = try XCTUnwrap(list, "Should parse: \(desc)")
XCTAssertTrue(unwrappedList.atoms.count >= 1, "\(desc) should have atoms")
}
}
// MARK: - Medium Priority Missing Features Tests
func testMultipleIntegrals() throws {
// Test \iint, \iiint, \iiiint for multiple integrals
let testCases = [
("\\iint f(x,y) dx dy", "double integral"),
("\\iiint f(x,y,z) dx dy dz", "triple integral"),
("\\iiiint f(w,x,y,z) dw dx dy dz", "quadruple integral"),
("\\iint_{D} f(x,y) dA", "double integral with limits")
]
for (latex, desc) in testCases {
var error: NSError? = nil
let list = MTMathListBuilder.build(fromString: latex, error: &error)
if let err = error {
XCTFail("ERROR: \(err.localizedDescription)")
} else if list == nil {
XCTFail("List is nil but no error")
}
let unwrappedList = try XCTUnwrap(list, "Should parse: \(desc)")
XCTAssertNil(error, "Should not error on \(desc): \(error?.localizedDescription ?? "")")
XCTAssertTrue(unwrappedList.atoms.count >= 1, "\(desc) should have atoms")
// Verify we have a large operator (integral) in the list
var foundOperator = false
for atom in unwrappedList.atoms {
if atom.type == .largeOperator {
foundOperator = true
break
}
}
XCTAssertTrue(foundOperator, "\(desc) should contain a large operator (integral)")
}
}
func testContinuedFractions() throws {
// Test \cfrac for continued fractions (already added but verify)
let testCases = [
("\\cfrac{1}{2}", "simple cfrac"),
("a_0 + \\cfrac{1}{a_1 + \\cfrac{1}{a_2}}", "nested cfrac"),
("\\cfrac{x^2}{y + \\cfrac{1}{z}}", "cfrac with expressions")
]
for (latex, desc) in testCases {
var error: NSError? = nil
let list = MTMathListBuilder.build(fromString: latex, error: &error)
// cfrac might be implemented, let's check
if list != nil && error == nil {
let unwrappedList = try XCTUnwrap(list, "Should parse: \(desc)")
XCTAssertTrue(unwrappedList.atoms.count >= 1, "\(desc) should have atoms")
} else {
throw XCTSkip("\\cfrac may have issues: \(desc). Error: \(error?.localizedDescription ?? "nil result")")
}
}
}
func testDisplayStyleFraction() throws {
// Test \dfrac - display-style fraction
let str = "\\dfrac{1}{2}"
var error: NSError? = nil
let list = MTMathListBuilder.build(fromString: str, error: &error)
let desc = "Error for string: \(str)"
XCTAssertNil(error, desc)
let unwrappedList = try XCTUnwrap(list, desc)
XCTAssertEqual(unwrappedList.atoms.count, 1, desc)
let frac = try XCTUnwrap(unwrappedList.atoms[0] as? MTFraction, desc)
XCTAssertEqual(frac.type, .fraction, desc)
XCTAssertTrue(frac.hasRule, desc)
// Check numerator
let numerator = try XCTUnwrap(frac.numerator, desc)
XCTAssertTrue(numerator.atoms.count >= 1, "Numerator should have at least style atom")
// First atom should be displaystyle
if numerator.atoms.count > 1 {
let styleAtom = numerator.atoms[0] as? MTMathStyle
XCTAssertNotNil(styleAtom, "First atom should be style atom")
XCTAssertEqual(styleAtom?.style, .display, "Should be display style")
}
// Check denominator
let denominator = try XCTUnwrap(frac.denominator, desc)
XCTAssertTrue(denominator.atoms.count >= 1, "Denominator should have at least style atom")
if denominator.atoms.count > 1 {
let styleAtom = denominator.atoms[0] as? MTMathStyle
XCTAssertNotNil(styleAtom, "First atom should be style atom")
XCTAssertEqual(styleAtom?.style, .display, "Should be display style")
}
}
func testTextStyleFraction() throws {
// Test \tfrac - text-style fraction
let str = "\\tfrac{a}{b}"
var error: NSError? = nil
let list = MTMathListBuilder.build(fromString: str, error: &error)
let desc = "Error for string: \(str)"
XCTAssertNil(error, desc)
let unwrappedList = try XCTUnwrap(list, desc)
XCTAssertEqual(unwrappedList.atoms.count, 1, desc)
let frac = try XCTUnwrap(unwrappedList.atoms[0] as? MTFraction, desc)
XCTAssertEqual(frac.type, .fraction, desc)
XCTAssertTrue(frac.hasRule, desc)
// Check numerator
let numerator = try XCTUnwrap(frac.numerator, desc)
XCTAssertTrue(numerator.atoms.count >= 1, "Numerator should have at least style atom")
if numerator.atoms.count > 1 {
let styleAtom = numerator.atoms[0] as? MTMathStyle
XCTAssertNotNil(styleAtom, "First atom should be style atom")
XCTAssertEqual(styleAtom?.style, .text, "Should be text style")
}
// Check denominator
let denominator = try XCTUnwrap(frac.denominator, desc)
XCTAssertTrue(denominator.atoms.count >= 1, "Denominator should have at least style atom")
if denominator.atoms.count > 1 {
let styleAtom = denominator.atoms[0] as? MTMathStyle
XCTAssertNotNil(styleAtom, "First atom should be style atom")
XCTAssertEqual(styleAtom?.style, .text, "Should be text style")
}
}
func testDisplayAndTextStyleFractions() throws {
// Test the original LaTeX from the user's issue
let str = "y'=-\\dfrac{2}{x^{3}}"
var error: NSError? = nil
let list = MTMathListBuilder.build(fromString: str, error: &error)
let desc = "Error for string: \(str)"
XCTAssertNil(error, desc)
let unwrappedList = try XCTUnwrap(list, desc)
XCTAssertTrue(unwrappedList.atoms.count >= 4, "Should have y, ', =, -, and fraction")
// Find the fraction atom
var foundFraction = false
for atom in unwrappedList.atoms {
if atom.type == .fraction {
foundFraction = true
let frac = atom as! MTFraction
// Check that numerator has displaystyle
if let numerator = frac.numerator, numerator.atoms.count > 0 {
let firstAtom = numerator.atoms[0]
if let styleAtom = firstAtom as? MTMathStyle {
XCTAssertEqual(styleAtom.style, .display, "Should force display style")
}
}
break
}
}
XCTAssertTrue(foundFraction, "Should find fraction in the expression")
// Test nested dfrac and tfrac
let nestedStr = "\\dfrac{\\tfrac{a}{b}}{c}"
error = nil
let nestedList = MTMathListBuilder.build(fromString: nestedStr, error: &error)
XCTAssertNil(error, "Should parse nested dfrac/tfrac")
XCTAssertNotNil(nestedList, "Should parse nested dfrac/tfrac")
}
func testBoldsymbol() throws {
// Test \boldsymbol for bold Greek letters
let testCases = [
("\\boldsymbol{\\alpha}", "bold alpha"),
("\\boldsymbol{\\beta}", "bold beta"),
("\\boldsymbol{\\Gamma}", "bold Gamma"),
("\\mathbf{x} + \\boldsymbol{\\mu}", "mixed bold")
]
for (latex, desc) in testCases {
var error: NSError? = nil
let list = MTMathListBuilder.build(fromString: latex, error: &error)
if list == nil || error != nil {
throw XCTSkip("\\boldsymbol not implemented: \(desc). Error: \(error?.localizedDescription ?? "nil result")")
}
let unwrappedList = try XCTUnwrap(list, "Should parse: \(desc)")
XCTAssertTrue(unwrappedList.atoms.count >= 1, "\(desc) should have atoms")
}
}
func testStarredMatrices() throws {
// Test starred matrix environments with alignment
let testCases = [
("\\begin{pmatrix*}[r] 1 & 2 \\\\ 3 & 4 \\end{pmatrix*}", "pmatrix* right align"),
("\\begin{bmatrix*}[l] a & b \\\\ c & d \\end{bmatrix*}", "bmatrix* left align"),
("\\begin{vmatrix*}[c] x & y \\\\ z & w \\end{vmatrix*}", "vmatrix* center align"),
("\\begin{matrix*}[r] 10 & 20 \\\\ 30 & 40 \\end{matrix*}", "matrix* right align (no delimiters)")
]
for (latex, desc) in testCases {
var error: NSError? = nil
let list = MTMathListBuilder.build(fromString: latex, error: &error)
let unwrappedList = try XCTUnwrap(list, "Should parse: \(desc)")
XCTAssertNil(error, "Should not error on \(desc): \(error?.localizedDescription ?? "")")
XCTAssertTrue(unwrappedList.atoms.count >= 1, "\(desc) should have atoms")
// Verify we have a table structure
var foundTable = false
for atom in unwrappedList.atoms {
if atom.type == .table {
foundTable = true
break
}
// Check inside inner atoms (for matrices with delimiters)
if atom.type == .inner, let inner = atom as? MTInner, let innerList = inner.innerList {
for innerAtom in innerList.atoms {
if innerAtom.type == .table {
foundTable = true
break
}
}
}
}
XCTAssertTrue(foundTable, "\(desc) should contain a table structure")
}
}
func testSmallMatrix() throws {
// Test \smallmatrix for inline matrices
let testCases = [
("\\left( \\begin{smallmatrix} a & b \\\\ c & d \\end{smallmatrix} \\right)", "smallmatrix with delimiters"),
("A = \\left( \\begin{smallmatrix} 1 & 0 \\\\ 0 & 1 \\end{smallmatrix} \\right)", "identity in smallmatrix"),
("\\begin{smallmatrix} x \\\\ y \\end{smallmatrix}", "column vector in smallmatrix")
]
for (latex, desc) in testCases {
var error: NSError? = nil
let list = MTMathListBuilder.build(fromString: latex, error: &error)
let unwrappedList = try XCTUnwrap(list, "Should parse: \(desc)")
XCTAssertNil(error, "Should not error on \(desc): \(error?.localizedDescription ?? "")")
XCTAssertTrue(unwrappedList.atoms.count >= 1, "\(desc) should have atoms")
// Verify we have a table structure
var foundTable = false
for atom in unwrappedList.atoms {
if atom.type == .table {
foundTable = true
break
}
// Check inside inner atoms (for matrices with delimiters)
if atom.type == .inner, let inner = atom as? MTInner, let innerList = inner.innerList {
for innerAtom in innerList.atoms {
if innerAtom.type == .table {
foundTable = true
break
}
}
}
}
XCTAssertTrue(foundTable, "\(desc) should contain a table structure")
}
}
// func testPerformanceExample() throws {
// // This is an example of a performance test case.
// measure {
// // Put the code you want to measure the time of here.
// }
// }
}