def test_memoization(self): f = Foobar() memoized = f.memoize(f.expensive()) self.assertNodeOutput("$put(v0,$expensive())", memoized) self.assertNodeOutput("$put(v0,$expensive())", memoized) self.assertNodeOutput("$moo($put(v0,$expensive()),$get(v0))", f.moo(memoized, memoized))
def test_multiple_memoizations(self): f = Foobar() m1 = f.memoize(f.one()) m2 = f.memoize(f.two()) self.assertNodeOutput("$put(v0,$one())", m1) self.assertNodeOutput("$put(v0,$two())", m2) self.assertNodeOutput("$put(v0,$one())", m1) self.assertNodeOutput("$put(v0,$two())", m2) self.assertNodeOutput("$foo($put(v0,$one()),$bar($get(v0),$put(v1,$two())),$get(v1))", f.foo(m1, f.bar(m1, m2), m2))
def test_multiple_memoizations(self): f = Foobar() m1 = f.memoize(f.one()) m2 = f.memoize(f.two()) self.assertNodeOutput("$put(v0,$one())", m1) self.assertNodeOutput("$put(v0,$two())", m2) self.assertNodeOutput("$put(v0,$one())", m1) self.assertNodeOutput("$put(v0,$two())", m2) self.assertNodeOutput( "$foo($put(v0,$one()),$bar($get(v0),$put(v1,$two())),$get(v1))", f.foo(m1, f.bar(m1, m2), m2))
def test_memoization(self): f = Foobar() memoized = f.memoize(f.expensive()) self.assertNodeOutput("$put(v0,$expensive())", memoized) self.assertNodeOutput("$put(v0,$expensive())", memoized) self.assertNodeOutput("$moo($put(v0,$expensive()),$get(v0))", f.moo(memoized, memoized))
def test_nesting(self): f = Foobar() m1 = f.memoize(f.hmm()) m2 = f.memoize(m1.hmm(m1)) self.assertNodeOutput("$put(v0,$hmm($put(v1,$hmm()),$get(v1)))", m2)
def test_nesting(self): f = Foobar() m1 = f.memoize(f.hmm()) m2 = f.memoize(m1.hmm(m1)) self.assertNodeOutput("$put(v0,$hmm($put(v1,$hmm()),$get(v1)))", m2)
def test_memoize(self): f = Foobar() self.assertEqualIncludingHash(f.memoize(f.foo()), f.memoize(f.foo())) self.assertNotEqualIncludingHash(f.memoize(f.foo()), f.memoize(f.bar()))