added some tests

XCTAssertEqual?