From 1e76193acb2553067f99983dcdc2cc4bc0e20992 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Sun, 30 Oct 2011 12:08:41 -0400 Subject: [PATCH] Support modulo operation in more places. --- pymbolic/mapper/__init__.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/pymbolic/mapper/__init__.py b/pymbolic/mapper/__init__.py index a2d4ec5..a08d4d6 100644 --- a/pymbolic/mapper/__init__.py +++ b/pymbolic/mapper/__init__.py @@ -123,10 +123,8 @@ class CombineMapper(RecursiveMapper): self.rec(expr.numerator, *args), self.rec(expr.denominator, *args))) - def map_floor_div(self, expr, *args): - return self.combine(( - self.rec(expr.numerator, *args), - self.rec(expr.denominator, *args))) + map_floor_div = map_quotient + map_remainder = map_quotient def map_power(self, expr, *args): return self.combine(( @@ -206,6 +204,7 @@ class IdentityMapperBase(object): self.rec(expr.denominator, *args)) map_floor_div = map_quotient + map_remainder = map_quotient def map_power(self, expr, *args): return expr.__class__(self.rec(expr.base, *args), -- GitLab