Converting tests to XCTest