Add Test Code

using XCTest