Converted to XCTest. Easier than I thought.