@@ -147,84 +147,84 @@ final class ComplexTests: XCTestCase {
147
147
let pb : ( Complex < Float > ) -> Complex < Float > = pullback ( at: Complex < Float > ( real: 2 , imaginary: 3 ) ) { x in
148
148
return x + Complex < Float > ( real: 5 , imaginary: 6 )
149
149
}
150
- expectEqual ( pb ( Complex ( real: 1 , imaginary: 1 ) ) , Complex < Float > ( real: 1 , imaginary: 1 ) )
150
+ XCTAssertEqual ( pb ( Complex ( real: 1 , imaginary: 1 ) ) , Complex < Float > ( real: 1 , imaginary: 1 ) )
151
151
}
152
152
153
153
func testVjpSubtract( ) {
154
154
let pb : ( Complex < Float > ) -> Complex < Float > = pullback ( at: Complex < Float > ( real: 2 , imaginary: 3 ) ) { x in
155
155
return Complex < Float > ( real: 5 , imaginary: 6 ) - x
156
156
}
157
- expectEqual ( pb ( Complex ( real: 1 , imaginary: 1 ) ) , Complex < Float > ( real: - 1 , imaginary: - 1 ) )
157
+ XCTAssertEqual ( pb ( Complex ( real: 1 , imaginary: 1 ) ) , Complex < Float > ( real: - 1 , imaginary: - 1 ) )
158
158
}
159
159
160
160
func testVjpMultiply( ) {
161
161
let pb : ( Complex < Float > ) -> Complex < Float > = pullback ( at: Complex < Float > ( real: 2 , imaginary: 3 ) ) { x in
162
162
return x * x
163
163
}
164
- expectEqual ( pb ( Complex ( real: 1 , imaginary: 0 ) ) , Complex < Float > ( real: 4 , imaginary: 6 ) )
165
- expectEqual ( pb ( Complex ( real: 0 , imaginary: 1 ) ) , Complex < Float > ( real: - 6 , imaginary: 4 ) )
166
- expectEqual ( pb ( Complex ( real: 1 , imaginary: 1 ) ) , Complex < Float > ( real: - 2 , imaginary: 10 ) )
164
+ XCTAssertEqual ( pb ( Complex ( real: 1 , imaginary: 0 ) ) , Complex < Float > ( real: 4 , imaginary: 6 ) )
165
+ XCTAssertEqual ( pb ( Complex ( real: 0 , imaginary: 1 ) ) , Complex < Float > ( real: - 6 , imaginary: 4 ) )
166
+ XCTAssertEqual ( pb ( Complex ( real: 1 , imaginary: 1 ) ) , Complex < Float > ( real: - 2 , imaginary: 10 ) )
167
167
}
168
168
169
169
func testVjpDivide( ) {
170
170
let pb : ( Complex < Float > ) -> Complex < Float > = pullback ( at: Complex < Float > ( real: 20 , imaginary: - 4 ) ) { x in
171
171
return x / Complex < Float > ( real: 2 , imaginary: 2 )
172
172
}
173
- expectEqual ( pb ( Complex ( real: 1 , imaginary: 0 ) ) , Complex < Float > ( real: 0.25 , imaginary: - 0.25 ) )
174
- expectEqual ( pb ( Complex ( real: 0 , imaginary: 1 ) ) , Complex < Float > ( real: 0.25 , imaginary: 0.25 ) )
173
+ XCTAssertEqual ( pb ( Complex ( real: 1 , imaginary: 0 ) ) , Complex < Float > ( real: 0.25 , imaginary: - 0.25 ) )
174
+ XCTAssertEqual ( pb ( Complex ( real: 0 , imaginary: 1 ) ) , Complex < Float > ( real: 0.25 , imaginary: 0.25 ) )
175
175
}
176
176
177
177
func testVjpNegate( ) {
178
178
let pb : ( Complex < Float > ) -> Complex < Float > = pullback ( at: Complex < Float > ( real: 20 , imaginary: - 4 ) ) { x in
179
179
return - x
180
180
}
181
- expectEqual ( pb ( Complex ( real: 1 , imaginary: 0 ) ) , Complex < Float > ( real: - 1 , imaginary: 0 ) )
182
- expectEqual ( pb ( Complex ( real: 0 , imaginary: 1 ) ) , Complex < Float > ( real: 0 , imaginary: - 1 ) )
183
- expectEqual ( pb ( Complex ( real: 1 , imaginary: 1 ) ) , Complex < Float > ( real: - 1 , imaginary: - 1 ) )
181
+ XCTAssertEqual ( pb ( Complex ( real: 1 , imaginary: 0 ) ) , Complex < Float > ( real: - 1 , imaginary: 0 ) )
182
+ XCTAssertEqual ( pb ( Complex ( real: 0 , imaginary: 1 ) ) , Complex < Float > ( real: 0 , imaginary: - 1 ) )
183
+ XCTAssertEqual ( pb ( Complex ( real: 1 , imaginary: 1 ) ) , Complex < Float > ( real: - 1 , imaginary: - 1 ) )
184
184
}
185
185
186
186
func testVjpComplexConjugate( ) {
187
187
let pb : ( Complex < Float > ) -> Complex < Float > = pullback ( at: Complex < Float > ( real: 20 , imaginary: - 4 ) ) { x in
188
188
return x. complexConjugate ( )
189
189
}
190
- expectEqual ( pb ( Complex ( real: 1 , imaginary: 0 ) ) , Complex < Float > ( real: - 1 , imaginary: 0 ) )
191
- expectEqual ( pb ( Complex ( real: 0 , imaginary: 1 ) ) , Complex < Float > ( real: 0 , imaginary: - 1 ) )
192
- expectEqual ( pb ( Complex ( real: 1 , imaginary: 1 ) ) , Complex < Float > ( real: - 1 , imaginary: - 1 ) )
190
+ XCTAssertEqual ( pb ( Complex ( real: 1 , imaginary: 0 ) ) , Complex < Float > ( real: - 1 , imaginary: 0 ) )
191
+ XCTAssertEqual ( pb ( Complex ( real: 0 , imaginary: 1 ) ) , Complex < Float > ( real: 0 , imaginary: - 1 ) )
192
+ XCTAssertEqual ( pb ( Complex ( real: 1 , imaginary: 1 ) ) , Complex < Float > ( real: - 1 , imaginary: - 1 ) )
193
193
}
194
194
195
195
func testVjpAdding( ) {
196
196
let pb : ( Complex < Float > ) -> Complex < Float > = pullback ( at: Complex < Float > ( real: 20 , imaginary: - 4 ) ) { x in
197
197
return x. adding ( real: 5 )
198
198
}
199
- expectEqual ( pb ( Complex ( real: 1 , imaginary: 0 ) ) , Complex < Float > ( real: 1 , imaginary: 0 ) )
200
- expectEqual ( pb ( Complex ( real: 0 , imaginary: 1 ) ) , Complex < Float > ( real: 0 , imaginary: 1 ) )
201
- expectEqual ( pb ( Complex ( real: 1 , imaginary: 1 ) ) , Complex < Float > ( real: 1 , imaginary: 1 ) )
199
+ XCTAssertEqual ( pb ( Complex ( real: 1 , imaginary: 0 ) ) , Complex < Float > ( real: 1 , imaginary: 0 ) )
200
+ XCTAssertEqual ( pb ( Complex ( real: 0 , imaginary: 1 ) ) , Complex < Float > ( real: 0 , imaginary: 1 ) )
201
+ XCTAssertEqual ( pb ( Complex ( real: 1 , imaginary: 1 ) ) , Complex < Float > ( real: 1 , imaginary: 1 ) )
202
202
}
203
203
204
204
func testVjpAdding( ) {
205
205
let pb : ( Complex < Float > ) -> Complex < Float > = pullback ( at: Complex < Float > ( real: 20 , imaginary: - 4 ) ) { x in
206
206
return x. adding ( imaginary: 5 )
207
207
}
208
- expectEqual ( pb ( Complex ( real: 1 , imaginary: 0 ) ) , Complex < Float > ( real: 1 , imaginary: 0 ) )
209
- expectEqual ( pb ( Complex ( real: 0 , imaginary: 1 ) ) , Complex < Float > ( real: 0 , imaginary: 1 ) )
210
- expectEqual ( pb ( Complex ( real: 1 , imaginary: 1 ) ) , Complex < Float > ( real: 1 , imaginary: 1 ) )
208
+ XCTAssertEqual ( pb ( Complex ( real: 1 , imaginary: 0 ) ) , Complex < Float > ( real: 1 , imaginary: 0 ) )
209
+ XCTAssertEqual ( pb ( Complex ( real: 0 , imaginary: 1 ) ) , Complex < Float > ( real: 0 , imaginary: 1 ) )
210
+ XCTAssertEqual ( pb ( Complex ( real: 1 , imaginary: 1 ) ) , Complex < Float > ( real: 1 , imaginary: 1 ) )
211
211
}
212
212
213
213
func testVjpSubtracting( ) {
214
214
let pb : ( Complex < Float > ) -> Complex < Float > = pullback ( at: Complex < Float > ( real: 20 , imaginary: - 4 ) ) { x in
215
215
return x. subtracting ( real: 5 )
216
216
}
217
- expectEqual ( pb ( Complex ( real: 1 , imaginary: 0 ) ) , Complex < Float > ( real: 1 , imaginary: 0 ) )
218
- expectEqual ( pb ( Complex ( real: 0 , imaginary: 1 ) ) , Complex < Float > ( real: 0 , imaginary: 1 ) )
219
- expectEqual ( pb ( Complex ( real: 1 , imaginary: 1 ) ) , Complex < Float > ( real: 1 , imaginary: 1 ) )
217
+ XCTAssertEqual ( pb ( Complex ( real: 1 , imaginary: 0 ) ) , Complex < Float > ( real: 1 , imaginary: 0 ) )
218
+ XCTAssertEqual ( pb ( Complex ( real: 0 , imaginary: 1 ) ) , Complex < Float > ( real: 0 , imaginary: 1 ) )
219
+ XCTAssertEqual ( pb ( Complex ( real: 1 , imaginary: 1 ) ) , Complex < Float > ( real: 1 , imaginary: 1 ) )
220
220
}
221
221
222
222
func testVjpSubtracting( ) {
223
223
let pb : ( Complex < Float > ) -> Complex < Float > = pullback ( at: Complex < Float > ( real: 20 , imaginary: - 4 ) ) { x in
224
224
return x. subtracting ( imaginary: 5 )
225
225
}
226
- expectEqual ( pb ( Complex ( real: 1 , imaginary: 0 ) ) , Complex < Float > ( real: 1 , imaginary: 0 ) )
227
- expectEqual ( pb ( Complex ( real: 0 , imaginary: 1 ) ) , Complex < Float > ( real: 0 , imaginary: 1 ) )
228
- expectEqual ( pb ( Complex ( real: 1 , imaginary: 1 ) ) , Complex < Float > ( real: 1 , imaginary: 1 ) )
226
+ XCTAssertEqual ( pb ( Complex ( real: 1 , imaginary: 0 ) ) , Complex < Float > ( real: 1 , imaginary: 0 ) )
227
+ XCTAssertEqual ( pb ( Complex ( real: 0 , imaginary: 1 ) ) , Complex < Float > ( real: 0 , imaginary: 1 ) )
228
+ XCTAssertEqual ( pb ( Complex ( real: 1 , imaginary: 1 ) ) , Complex < Float > ( real: 1 , imaginary: 1 ) )
229
229
}
230
230
}
0 commit comments