Add XCTest compatibility