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()))
