Esempio n. 1
0
def reify_object(o, s):
    """ Reify a Python object with a substitution

    >>> from logpy.unifymore import reify_object
    >>> from logpy import var
    >>> class Foo(object):
    ...     def __init__(self, a, b):
    ...         self.a = a
    ...         self.b = b
    ...     def __str__(self):
    ...         return "Foo(%s, %s)"%(str(self.a), str(self.b))

    >>> x = var('x')
    >>> f = Foo(1, x)
    >>> print f
    Foo(1, ~x)
    >>> print reify_object(f, {x: 2})
    Foo(1, 2)
    """

    obj = object.__new__(type(o))
    d = reify_dict(o.__dict__, s)
    if d == o.__dict__:
        return o
    obj.__dict__.update(d)
    return obj
Esempio n. 2
0
def reify_object_attrs(o, s, attrs):
    """ Reify only certain attributes of a Python object

    >>> from logpy.unifymore import reify_object_attrs
    >>> from logpy import var
    >>> class Foo(object):
    ...     def __init__(self, a, b):
    ...         self.a = a
    ...         self.b = b
    ...     def __str__(self):
    ...         return "Foo(%s, %s)"%(str(self.a), str(self.b))

    >>> x = var('x')
    >>> y = var('y')
    >>> f = Foo(x, y)
    >>> print f
    Foo(~x, ~y)
    >>> print reify_object_attrs(f, {x: 1, y: 2}, ['a', 'b'])
    Foo(1, 2)
    >>> print reify_object_attrs(f, {x: 1, y: 2}, ['a'])
    Foo(1, ~y)

    This function is meant to be partially specialized

    >>> from functools import partial
    >>> reify_Foo_a = partial(reify_object_attrs, attrs=['a'])

    attrs contains the list of attributes which participate in reificiation
    """
    obj = object.__new__(type(o))
    d = dict(zip(attrs, map(o.__dict__.get, attrs)))  # dict with attrs
    d2 = reify_dict(d, s)  # reified attr dict
    if d2 == d:
        return o
    obj.__dict__.update(o.__dict__)  # old dict
    obj.__dict__.update(d2)  # update w/ reified vals
    return obj
Esempio n. 3
0
def test_reify_dict():
    x, y = var(), var()
    s = {x: 2, y: 4}
    e = {1: x, 3: {5: y}}
    assert reify_dict(e, s) == {1: 2, 3: {5: 4}}