be smarter about NaN/Inf in roots()

rakudo · Sep 30, 2009 · 3566baf · 3566baf
1 parent 65af3ca
commit 3566baf
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/setting/Complex.pm b/src/setting/Complex.pm
@@ -62,7 +62,9 @@ class Complex {
 
     multi method roots($n is copy) {
         my ($mag, $angle) = @.polar;
-	return NaN if $n < 1;
+	return NaN  if $n < 1;
+	return self if $n == 1;
+	return NaN  if $!re|$!im ~~  Inf|NaN|-Inf; 
 	$n = $n.Int;
         $mag **= 1/$n;
         (^$n).map: { $mag.unpolar( ($angle + $_ * 2 * pi) / $n) };