search

HPAC / matchpy

A library for pattern matching on symbolic expressions in Python.
MIT License
164 stars 25 forks source link

'Integer' object is not iterable in `ManyToOneReplacer` #23

Closed arihantparsoya closed 7 years ago

arihantparsoya commented 7 years ago

Pattern is matching but .replace in ManyToOneReplacer is showing error.

Code

import matchpy
Pattern, ReplacementRule, ManyToOneReplacer = matchpy.Pattern, matchpy.ReplacementRule, matchpy.ManyToOneReplacer

from matchpy import replace_all, is_match, Wildcard
from sympy.integrals import Integral
from sympy import Symbol, Pow, cacheit, Basic, S, Add, Mul, srepr, Or, And
from matchpy.expressions.functions import register_operation_iterator, register_operation_factory
from matchpy import Operation, CommutativeOperation, AssociativeOperation, OneIdentityOperation, match, CustomConstraint

class WC(Wildcard, Symbol):
    def __init__(self, min_length, fixed_size, variable_name=None, optional=None, **assumptions):
        Wildcard.__init__(self, min_length, fixed_size, variable_name, optional)

    def __new__(cls, min_length, fixed_size, variable_name=None, optional=None, **assumptions):
        cls._sanitize(assumptions, cls)
        return WC.__xnew__(cls, min_length, fixed_size, variable_name, optional, **assumptions)

    def __getnewargs__(self):
        return (self.min_length, self.fixed_size, self.variable_name, self.optional)

    @staticmethod
    def __xnew__(cls, min_length, fixed_size, variable_name=None, optional=None, **assumptions):
        obj = Symbol.__xnew__(cls, variable_name, **assumptions)
        return obj

    def _hashable_content(self):
        return super()._hashable_content() + (self.min_count, self.fixed_size, self.variable_name, self.optional)

Operation.register(Integral)
register_operation_iterator(Integral, lambda a: (a._args[0],) + a._args[1], lambda a: len((a._args[0],) + a._args[1]))

Operation.register(Pow)
OneIdentityOperation.register(Pow)
register_operation_iterator(Pow, lambda a: a._args, lambda a: len(a._args))

Operation.register(Add)
OneIdentityOperation.register(Add)
CommutativeOperation.register(Add)
AssociativeOperation.register(Add)
register_operation_iterator(Add, lambda a: a._args, lambda a: len(a._args))

Operation.register(Mul)
OneIdentityOperation.register(Mul)
CommutativeOperation.register(Mul)
AssociativeOperation.register(Mul)
register_operation_iterator(Mul, lambda a: a._args, lambda a: len(a._args))

def sympy_op_factory(old_operation, new_operands, variable_name):
     return type(old_operation)(*new_operands)

register_operation_factory(Basic, sympy_op_factory)

m_ = WC(1, True, 'm')
x_ = WC(1, True, 'x')
c_ = WC(1, True, 'c')
n_ = WC(1, True, 'n')
x = Symbol('x')
b = Symbol('b')
a = Symbol('a')
c = Symbol('c')
d = Symbol('d')
m = Symbol('m')
n = Symbol('n')

subject = Integral((b*x)**m*(d*x + S(2))**n, x)

rubi = ManyToOneReplacer()

pattern45 = Pattern(Integral(Mul(Pow(Mul(WC(1, True, 'b', S('1')), x_), m_), Pow(Add(c_, Mul(WC(1, True, 'd', S('1')), x_)), n_)), x_))
rule45 = ReplacementRule(pattern45, lambda d, x, b, c, m, n : Mul(Pow(c, n), Mul(Pow(Mul(b, x), Add(m, S('1'))), Pow(Mul(b, Add(m, S('1'))), S('-1'))), Hypergeometric2F1(Mul(S('-1'), n), Add(m, S('1')), Add(m, S('2')), Mul(Mul(S('-1'), d), Mul(x, Pow(c, S('-1')))))))
rubi.add(rule45)

print(is_match(subject, pattern45))
print(rubi.replace(subject))

Error

True
Traceback (most recent call last):
  File "r.py", line 70, in <module>
    print(rubi.replace(subject))
  File "/Users/parsoyaarihant/anaconda/lib/python3.6/site-packages/matchpy-0.5.dev56+g9b820f8-py3.6.egg/matchpy/matching/many_to_one.py", line 796, in replace
  File "/Users/parsoyaarihant/anaconda/lib/python3.6/site-packages/matchpy-0.5.dev56+g9b820f8-py3.6.egg/matchpy/matching/many_to_one.py", line 99, in __iter__
  File "/Users/parsoyaarihant/anaconda/lib/python3.6/site-packages/matchpy-0.5.dev56+g9b820f8-py3.6.egg/matchpy/matching/many_to_one.py", line 148, in _match
  File "/Users/parsoyaarihant/anaconda/lib/python3.6/site-packages/matchpy-0.5.dev56+g9b820f8-py3.6.egg/matchpy/matching/many_to_one.py", line 162, in _match_transition
  File "/Users/parsoyaarihant/anaconda/lib/python3.6/site-packages/matchpy-0.5.dev56+g9b820f8-py3.6.egg/matchpy/matching/many_to_one.py", line 310, in _match_regular_operation
  File "/Users/parsoyaarihant/anaconda/lib/python3.6/site-packages/matchpy-0.5.dev56+g9b820f8-py3.6.egg/matchpy/matching/many_to_one.py", line 209, in _check_transition
  File "/Users/parsoyaarihant/anaconda/lib/python3.6/site-packages/matchpy-0.5.dev56+g9b820f8-py3.6.egg/matchpy/matching/many_to_one.py", line 148, in _match
  File "/Users/parsoyaarihant/anaconda/lib/python3.6/site-packages/matchpy-0.5.dev56+g9b820f8-py3.6.egg/matchpy/matching/many_to_one.py", line 160, in _match_transition
  File "/Users/parsoyaarihant/anaconda/lib/python3.6/site-packages/matchpy-0.5.dev56+g9b820f8-py3.6.egg/matchpy/matching/many_to_one.py", line 281, in _match_commutative_operation
  File "/Users/parsoyaarihant/anaconda/lib/python3.6/site-packages/matchpy-0.5.dev56+g9b820f8-py3.6.egg/matchpy/matching/many_to_one.py", line 849, in add_subject
  File "/Users/parsoyaarihant/anaconda/lib/python3.6/site-packages/matchpy-0.5.dev56+g9b820f8-py3.6.egg/matchpy/matching/many_to_one.py", line 839, in get_match_iter
  File "/Users/parsoyaarihant/anaconda/lib/python3.6/site-packages/matchpy-0.5.dev56+g9b820f8-py3.6.egg/matchpy/matching/many_to_one.py", line 148, in _match
  File "/Users/parsoyaarihant/anaconda/lib/python3.6/site-packages/matchpy-0.5.dev56+g9b820f8-py3.6.egg/matchpy/matching/many_to_one.py", line 162, in _match_transition
  File "/Users/parsoyaarihant/anaconda/lib/python3.6/site-packages/matchpy-0.5.dev56+g9b820f8-py3.6.egg/matchpy/matching/many_to_one.py", line 310, in _match_regular_operation
  File "/Users/parsoyaarihant/anaconda/lib/python3.6/site-packages/matchpy-0.5.dev56+g9b820f8-py3.6.egg/matchpy/matching/many_to_one.py", line 209, in _check_transition
  File "/Users/parsoyaarihant/anaconda/lib/python3.6/site-packages/matchpy-0.5.dev56+g9b820f8-py3.6.egg/matchpy/matching/many_to_one.py", line 148, in _match
  File "/Users/parsoyaarihant/anaconda/lib/python3.6/site-packages/matchpy-0.5.dev56+g9b820f8-py3.6.egg/matchpy/matching/many_to_one.py", line 160, in _match_transition
  File "/Users/parsoyaarihant/anaconda/lib/python3.6/site-packages/matchpy-0.5.dev56+g9b820f8-py3.6.egg/matchpy/matching/many_to_one.py", line 282, in _match_commutative_operation
  File "/Users/parsoyaarihant/anaconda/lib/python3.6/site-packages/matchpy-0.5.dev56+g9b820f8-py3.6.egg/matchpy/matching/many_to_one.py", line 865, in match
TypeError: 'Integer' object is not iterable