Complex.ToString()

#21
swharden · Jul 15, 2020 · f4c2a53 · f4c2a53
1 parent f7e5de5
commit f4c2a53
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/src/FftSharp.Tests/ComplexTests.cs b/src/FftSharp.Tests/ComplexTests.cs
@@ -0,0 +1,20 @@
+using NUnit.Framework;
+using System;
+using System.Collections.Generic;
+using System.Text;
+
+namespace FftSharp.Tests
+{
+    class ComplexTests
+    {
+        [Test]
+        public void Test_Complex_ToString()
+        {
+            Assert.AreEqual("0+0j", new FftSharp.Complex(0, 0).ToString());
+            Assert.AreEqual("3+5j", new FftSharp.Complex(3, 5).ToString());
+            Assert.AreEqual("-3-5j", new FftSharp.Complex(-3, -5).ToString());
+            Assert.AreEqual("3-5j", new FftSharp.Complex(3, -5).ToString());
+            Assert.AreEqual("-3+5j", new FftSharp.Complex(-3, 5).ToString());
+        }
+    }
+}
diff --git a/src/FftSharp/Complex.cs b/src/FftSharp/Complex.cs
@@ -17,6 +17,14 @@ public Complex(double real, double imaginary)
             Imaginary = imaginary;
         }
 
+        public override string ToString()
+        {
+            if (Imaginary < 0)
+                return $"{Real}-{-Imaginary}j";
+            else
+                return $"{Real}+{Imaginary}j";
+        }
+
         public static Complex operator +(Complex a, Complex b)
         {
             return new Complex(a.Real + b.Real, a.Imaginary + b.Imaginary);