feat: i extension property

KatieUmbra · Nov 10, 2023 · 59f4d5c · 59f4d5c
1 parent ad93430
commit 59f4d5c
Show file tree

Hide file tree

Showing 3 changed files with 23 additions and 3 deletions.
diff --git a/.gradle/buildOutputCleanup/buildOutputCleanup.lock b/.gradle/buildOutputCleanup/buildOutputCleanup.lock
diff --git a/src/main/kotlin/compxclib/operators/Operators.kt b/src/main/kotlin/compxclib/operators/Operators.kt
@@ -79,4 +79,13 @@ operator fun Number.rem(complexNumber: ComplexNumber): ComplexNumber {
 @Suppress("unused")
 infix fun Number.pow(to: ComplexNumber): ComplexNumber {
     return exp(ln(this.toDouble()) * to)
-}
+}
+
+// Extension for a number
+/**
+ * `i` extension property for numbers that return an imaginary number
+ * @since Version 1.2
+ */
+@Suppress("unused")
+val Number.i: ComplexNumber
+    get() = ComplexNumber(0, this)
diff --git a/src/test/kotlin/compxclib/FunctionTest.kt b/src/test/kotlin/compxclib/FunctionTest.kt
@@ -1,10 +1,13 @@
 package compxclib
 
 import compxclib.enums.SqrtBranch
-import kotlin.test.*
-import kotlin.math.cosh
 import compxclib.functions.*
+import compxclib.operators.i
+import compxclib.operators.plus
 import compxclib.operators.times
+import kotlin.math.cosh
+import kotlin.test.Test
+import kotlin.test.assertEquals
 
 class FunctionTest {
 
@@ -116,4 +119,12 @@ class FunctionTest {
             arccsch(ComplexNumber(-1, -1))
         )
     }
+
+    @Test
+    fun iTest() {
+        assertEquals(
+            ComplexNumber(5, 6),
+            5 + 6.i
+        )
+    }
 }