forked from swiftlang/swift
-
Notifications
You must be signed in to change notification settings - Fork 0
/
simple_math.swift
51 lines (45 loc) · 1.13 KB
/
simple_math.swift
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
// RUN: %target-run-simple-swift
// REQUIRES: executable_test
import StdlibUnittest
#if os(macOS)
import Darwin.C
#else
import Glibc
#endif
var SimpleMathTests = TestSuite("SimpleMath")
SimpleMathTests.test("Arithmetics") {
let dfoo1 = #gradient({ (x: Float, y: Float) -> Float in
return x * y
})
expectEqual((4, 3), dfoo1(3, 4))
let dfoo2 = #gradient({ (x: Float, y: Float) -> Float in
return -x * y
})
expectEqual((-4, -3), dfoo2(3, 4))
let dfoo3 = #gradient({ (x: Float, y: Float) -> Float in
return -x + y
})
expectEqual((-1, 1), dfoo3(3, 4))
}
SimpleMathTests.test("Fanout") {
let dfoo1 = #gradient({ (x: Float) -> Float in
x - x
})
expectEqual(0, dfoo1(100))
let dfoo2 = #gradient({ (x: Float) -> Float in
x + x
})
expectEqual(2, dfoo2(100))
let dfoo3 = #gradient({ (x: Float, y: Float) -> Float in
x + x + x * y
})
expectEqual((4, 3), dfoo3(3, 2))
}
SimpleMathTests.test("FunctionCall") {
func foo(_ x: Float, _ y: Float) -> Float {
return 3 * x + { $0 * 3 }(3) * y
}
expectEqual((3, 9), #gradient(foo)(3, 4))
expectEqual(3, #gradient(foo, wrt: .0)(3, 4))
}
runAllTests()