Test complex slices.

Author: NoamK-CR

_examples/slices/slices.go

@@ -4,7 +4,10 @@
 
 package slices
 
-import "fmt"
+import (
+	"fmt"
+	"math/cmplx"
+)
 
 func IntSum(s []int) int {
 	sum := 0
@@ -28,6 +31,8 @@ type SliceInt16 []int16
 type SliceInt32 []int32
 type SliceInt64 []int64
 
+type SliceComplex []complex128
+
 type SliceIface []interface{}
 
 type S struct {
@@ -47,3 +52,11 @@ func PrintSSlice(ss []*S) {
 func PrintS(s *S) {
 	fmt.Printf("%v\n", s.Name)
 }
+
+func CmplxSqrt(arr SliceComplex) SliceComplex {
+	res := make([]complex128, len(arr))
+	for i, el := range arr {
+		res[i] = cmplx.Sqrt(el)
+	}
+	return res
+}

_examples/slices/test.py

@@ -3,6 +3,8 @@
 # license that can be found in the LICENSE file.
 
 from __future__ import print_function
+import math
+import random
 import slices, go
 
 a = [1,2,3,4]
@@ -35,4 +37,11 @@
 slices.PrintS(ss[0])
 slices.PrintS(ss[1])
 
+cmplx = slices.SliceComplex([(random.random() + random.random() * 1j) for _ in range(16)])
+sqrts = slices.CmplxSqrt(cmplx)
+for root, orig in zip(sqrts, cmplx):
+    root_squared = root * root
+    assert math.isclose(root_squared.real, orig.real)
+    assert math.isclose(root_squared.imag, orig.imag)
+
 print("OK")