polymod yields 1 more result than args unless inf

rakudo · Feb 14, 2015 · 0e61279 · 0e61279
1 parent 874caf6
commit 0e61279
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/src/core/Int.pm b/src/core/Int.pm
@@ -62,10 +62,11 @@ my class Int does Real { # declared in BOOTSTRAP
     # If do-not-want, user should cast invocant to proper domain.
     method polymod(Int:D: *@mods) {
         my $more = self;
+        my $inf = @mods.elems == Inf;
         fail X::OutOfRange.new(what => 'invocant to polymod', got => $more, range => "0..*") if $more < 0;
         gather {
             for @mods -> $mod {
-                last unless $more;
+                last if $inf and not $more;
                 fail X::Numeric::DivideByZero.new(using => 'polymod') unless $mod;
                 take $more mod $mod;
                 $more div= $mod;

diff --git a/src/core/Real.pm b/src/core/Real.pm
@@ -66,10 +66,11 @@ my role Real does Numeric {
 
     method polymod(Real:D: *@mods) {
         my $more = self;
+        my $inf = @mods.elems == Inf;
         fail X::OutOfRange.new(what => 'invocant to polymod', got => $more, range => "0..*") if $more < 0;
         gather {
             for @mods -> $mod {
-                last unless $more;
+                last if $inf and not $more;
                 fail X::Numeric::DivideByZero.new(using => 'polymod') unless $mod;
                 take my $rem = $more % $mod;
                 $more -= $rem;