Commit 9b6d7d5cac1af3217599e9fbd23fc83a9e536f76
2021-12-01T22:10:29
Add Swift Test case for libtommath
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 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100
diff --git a/demo/tommath_tests.swift b/demo/tommath_tests.swift
new file mode 100644
index 0000000..fd254da
--- /dev/null
+++ b/demo/tommath_tests.swift
@@ -0,0 +1,94 @@
+import XCTest
+import libtommath
+
+/* ---> Basic Manipulations <--- */
+
+extension mp_int {
+ var isZero: Bool { used == 0 }
+ var isNeg: Bool { sign == MP_NEG }
+ var isEven: Bool { used == 0 || (dp[0] & 1) == 0 }
+ var isOdd: Bool { !isEven }
+}
+
+func mp_get_u32(_ a: UnsafePointer<mp_int>) -> UInt32 {
+ return UInt32(bitPattern: mp_get_i32(a))
+}
+
+class LibTommathTests: XCTestCase {
+
+ override func setUpWithError() throws {
+ // Put setup code here. This method is called before the invocation of each test method in the class.
+ }
+
+ override func tearDownWithError() throws {
+ // Put teardown code here. This method is called after the invocation of each test method in the class.
+ }
+
+ func testTrivialStuff() throws {
+ var a = mp_int()
+ var b = mp_int()
+ var c = mp_int()
+ var d = mp_int()
+
+ XCTAssertEqual(mp_init(&a), MP_OKAY)
+ XCTAssertEqual(mp_init(&b), MP_OKAY)
+ XCTAssertEqual(mp_init(&c), MP_OKAY)
+ XCTAssertEqual(mp_init(&d), MP_OKAY)
+
+ defer {
+ mp_clear(&a)
+ mp_clear(&b)
+ mp_clear(&c)
+ mp_clear(&d)
+ }
+
+ XCTAssert(mp_error_to_string(MP_OKAY) != nil)
+
+ /* a: 0->5 */
+ mp_set(&a, 5)
+ /* a: 5-> b: -5 */
+ XCTAssertEqual(mp_neg(&a, &b), MP_OKAY)
+ XCTAssertEqual(mp_cmp(&a, &b), MP_GT)
+ XCTAssertEqual(mp_cmp(&b, &a), MP_LT)
+ XCTAssertTrue(b.isNeg)
+ /* a: 5-> a: -5 */
+ var t = a // Fix compiler error: Overlapping accesses to 'a', but modification requires exclusive access; consider copying to a local variable
+ XCTAssertEqual(mp_neg(&t, &a), MP_OKAY)
+ XCTAssertEqual(mp_cmp(&b, &a), MP_EQ)
+ XCTAssertTrue(a.isNeg)
+ /* a: -5-> b: 5 */
+ XCTAssertEqual(mp_abs(&a, &b), MP_OKAY)
+ XCTAssertTrue(!b.isNeg)
+ /* a: -5-> b: -4 */
+ XCTAssertEqual(mp_add_d(&a, 1, &b), MP_OKAY)
+ XCTAssertTrue(b.isNeg)
+ XCTAssertEqual(mp_get_i32(&b), -4)
+ XCTAssertEqual(mp_get_u32(&b), UInt32(bitPattern: -4))
+ XCTAssertEqual(mp_get_mag_u32(&b), 4)
+ /* a: -5-> b: 1 */
+ XCTAssertEqual(mp_add_d(&a, 6, &b), MP_OKAY)
+ XCTAssertEqual(mp_get_u32(&b), 1)
+ /* a: -5-> a: 1 */
+ t = a
+ XCTAssertEqual(mp_add_d(&t, 6, &a), MP_OKAY)
+ XCTAssertEqual(mp_get_u32(&a), 1)
+ mp_zero(&a);
+ /* a: 0-> a: 6 */
+ t = a
+ XCTAssertEqual(mp_add_d(&t, 6, &a), MP_OKAY)
+ XCTAssertEqual(mp_get_u32(&a), 6)
+
+ mp_set(&a, 42)
+ mp_set(&b, 1)
+ t = b
+ XCTAssertEqual(mp_neg(&t, &b), MP_OKAY)
+ mp_set(&c, 1)
+ XCTAssertEqual(mp_exptmod(&a, &b, &c, &d), MP_OKAY)
+
+ mp_set(&c, 7)
+ /* same here */
+ XCTAssertTrue(mp_exptmod(&a, &b, &c, &d) != MP_OKAY)
+
+ XCTAssertTrue(a.isEven != a.isOdd)
+ }
+}