Add some tests

Author: CodaFi

Tests/LLVMTests/ConstantSpec.swift

@@ -0,0 +1,92 @@
+import LLVM
+import XCTest
+import Foundation
+
+class ConstantSpec : XCTestCase {
+  func testConstants() {
+
+    XCTAssert(fileCheckOutput(of: .stderr, withPrefixes: ["SIGNEDCONST"]) {
+      // SIGNEDCONST: ; ModuleID = '[[ModuleName:ConstantTest]]'
+      // SIGNEDCONST-NEXT: source_filename = "[[ModuleName]]"
+      let module = Module(name: "ConstantTest")
+      let builder = IRBuilder(module: module)
+      // SIGNEDCONST: define void @main() {
+      let main = builder.addFunction("main",
+                                     type: FunctionType(argTypes: [],
+                                                        returnType: VoidType()))
+      let constant = IntType.int64.constant(42)
+
+      // SIGNEDCONST-NEXT: entry:
+      let entry = main.appendBasicBlock(named: "entry")
+      builder.positionAtEnd(of: entry)
+
+      // SIGNEDCONST-NOT: %{{[0-9]+}} = add i64 %%{{[0-9]+}}, %%{{[0-9]+}}
+      let val1 = builder.buildAdd(constant + constant, constant * constant)
+      // SIGNEDCONST-NOT: %{{[0-9]+}} = sub i64 %%{{[0-9]+}}, %%{{[0-9]+}}
+      let val2 = builder.buildSub(constant - constant, constant / constant)
+      // SIGNEDCONST-NOT: %{{[0-9]+}} = mul i64 %%{{[0-9]+}}, %%{{[0-9]+}}
+      let val3 = builder.buildMul(val1, val2)
+
+      // SIGNEDCONST-NEXT: ret i64 -1848
+      builder.buildRet(val3)
+      // SIGNEDCONST-NEXT: }
+      module.dump()
+    })
+
+    XCTAssert(fileCheckOutput(of: .stderr, withPrefixes: ["UNSIGNEDCONST"]) {
+      // UNSIGNEDCONST: ; ModuleID = '[[ModuleName:ConstantTest]]'
+      // UNSIGNEDCONST-NEXT: source_filename = "[[ModuleName]]"
+      let module = Module(name: "ConstantTest")
+      let builder = IRBuilder(module: module)
+      // UNSIGNEDCONST: define void @main() {
+      let main = builder.addFunction("main",
+                                     type: FunctionType(argTypes: [],
+                                                        returnType: VoidType()))
+      let constant = IntType.int64.constant(UInt64(42))
+
+      // UNSIGNEDCONST-NEXT: entry:
+      let entry = main.appendBasicBlock(named: "entry")
+      builder.positionAtEnd(of: entry)
+
+      // UNSIGNEDCONST-NOT: %{{[0-9]+}} = add i64 %%{{[0-9]+}}, %%{{[0-9]+}}
+      let val1 = builder.buildAdd(constant + constant, constant * constant)
+      // UNSIGNEDCONST-NOT: %{{[0-9]+}} = sub i64 %%{{[0-9]+}}, %%{{[0-9]+}}
+      let val2 = builder.buildSub(constant - constant, constant / constant)
+      // UNSIGNEDCONST-NOT: %{{[0-9]+}} = mul i64 %%{{[0-9]+}}, %%{{[0-9]+}}
+      let val3 = builder.buildMul(val1, val2)
+
+      // UNSIGNEDCONST-NEXT: ret i64 -1848
+      builder.buildRet(val3)
+      // UNSIGNEDCONST-NEXT: }
+      module.dump()
+    })
+
+    XCTAssert(fileCheckOutput(of: .stderr, withPrefixes: ["FLOATINGCONST"]) {
+      // FLOATINGCONST: ; ModuleID = '[[ModuleName:ConstantTest]]'
+      // FLOATINGCONST-NEXT: source_filename = "[[ModuleName]]"
+      let module = Module(name: "ConstantTest")
+      let builder = IRBuilder(module: module)
+      // FLOATINGCONST: define void @main() {
+      let main = builder.addFunction("main",
+                                     type: FunctionType(argTypes: [],
+                                                        returnType: VoidType()))
+      let constant = FloatType.double.constant(42.0)
+
+      // FLOATINGCONST-NEXT: entry:
+      let entry = main.appendBasicBlock(named: "entry")
+      builder.positionAtEnd(of: entry)
+
+      // FLOATINGCONST-NOT: %{{[0-9]+}} = add double %%{{[0-9]+}}, %%{{[0-9]+}}
+      let val1 = builder.buildAdd(constant + constant, constant * constant)
+      // FLOATINGCONST-NOT: %{{[0-9]+}} = sub double %%{{[0-9]+}}, %%{{[0-9]+}}
+      let val2 = builder.buildSub(constant - constant, constant / constant)
+      // FLOATINGCONST-NOT: %{{[0-9]+}} = mul double %%{{[0-9]+}}, %%{{[0-9]+}}
+      let val3 = builder.buildMul(val1, val2)
+
+      // FLOATINGCONST-NEXT: ret double -1.848000e+03
+      builder.buildRet(val3)
+      // FLOATINGCONST-NEXT: }
+      module.dump()
+    })
+  }
+}