def test_reusable_scope(self):
scope = let(a="tacos", b="soup", c="cake")
d = "godzilla"
with scope:
self.assertEquals(a, "tacos")
self.assertEquals(b, "soup")
self.assertEquals(c, "cake")
self.assertEquals(d, "godzilla")
a = "fajita"
b = "stew"
d = "mothra"
self.assertFalse("a" in locals())
self.assertFalse("b" in locals())
self.assertFalse("c" in locals())
self.assertTrue("d" in locals())
self.assertFalse("a" in globals())
self.assertFalse("b" in globals())
self.assertFalse("c" in globals())
self.assertFalse("d" in globals())
self.assertEquals(d, "mothra")
with scope:
self.assertEquals(a, "fajita")
self.assertEquals(b, "stew")
self.assertEquals(c, "cake")
self.assertEquals(d, "mothra")
def test_mismatch(self):
closer_frame = [None]
scope = let(a="pizza", b="beer")
scope.__enter__()
self.assertTrue(scope.in_use())
def closer():
closer_frame[0] = currentframe()
scope.__exit__(None, None, None)
# calling closer executes the scope exit inside closer's frame
# rather than this frame, which is a mismatch.
self.assertRaises(ScopeMismatch, closer)
# raise the exception again, but keep it for observation
try:
closer()
except ScopeMismatch as smm:
pass
# testing the correct fields were gathered
self.assertEquals(scope, smm.scope)
self.assertEquals(currentframe(), smm.frame)
self.assertEquals(closer_frame[0], smm.wrong_frame)
# we need to clean up that __enter__ so globals isn't a mess
# for later tests
scope.__exit__(None, None, None)
with scope.alias():
self.assertEquals(a, "pizza")
self.assertEquals(b, "beer")
def test_del_fast_local(self):
a = "tacos"
# testing `DELETE_NAME` opcode. Since `a` is already defined
# above when we push our scope `a` will be considered a fast
# local, and calling `del a` will trigger `DELETE_NAME`
with let(a="pizza") as scope:
self.assertEquals(a, "pizza")
del a
# this doesn't work -- I cannot make it fall-through,
# there's nothing that lets me catch the DELETE_NAME
# opcode's execution, and the f_locals.__delitem__ is
# only triggered when the locals() builtin is called
#self.assertEquals(a, "taco")
self.assertTrue("a" not in locals())
self.assertEquals(a, "tacos")
with scope:
self.assertEquals(a, "tacos")
#print locals()
self.assertEquals(a, "tacos")
def test_alias(self):
a = "hungry"
b = "thirsty"
# here we'll set up a helper function that will activate
# a given scope, and then assign the a and b values to the
# food and drink we supply.
def set_in_scope(my_scope, food, drink):
with my_scope:
a = food
b = drink
with let(a="pizza", b="beer") as scope:
# attempting to activate the existing scope prevents
# us from setting the rat poison
self.assertRaises(ScopeInUse, set_in_scope,
scope, "rat", "poison")
self.assertEquals(a, "pizza")
self.assertEquals(b, "beer")
# instead, we alias the existing active scope, and we can
# happily use it instead. When it exits, our scope is
# refreshed with any values that might have changed via
# the alias.
capture = scope.alias()
set_in_scope(capture, "tacos", "soda")
self.assertEquals(a, "tacos")
self.assertEquals(b, "soda")
self.assertEquals(a, "hungry")
self.assertEquals(b, "thirsty")
def test_nonlocal_del(self):
# weird pythonism: the statement `del a` WITHOUT a `global a`
# causes the compiler to create an empty fast local variable
# named `a` with a NULL value, which it then deletes (setting
# the value again to NULL. In other words, the del statement
# counts as an assignment in the eyes of the compiler.
global a, b
a = "remove me"
with let(a="tacos", b="soda") as my_scope:
self.assertTrue("a" in globals())
self.assertTrue("b" in globals())
del a
del b
self.assertTrue("a" not in my_scope)
self.assertTrue("b" not in my_scope)
self.assertEquals(a, "remove me")
del a
self.assertTrue("a" not in globals())
self.assertTrue("b" not in globals())
def test_simple_single(self):
a = "tacos"
b = "soda"
with let(a="fajita"):
self.assertEquals(a, "fajita")
self.assertEquals(b, "soda")
self.assertEquals(a, "tacos")
self.assertEquals(b, "soda")
def test_nested_assign(self):
"""
tests that direct assignment of a non-re-scoped variable correctly
sets the right valueq
"""
a = "tacos"
b = "soup"
c = "cake"
with let():
a = "fajita"
with let():
self.assertEquals(a, "fajita")
b = "stew"
self.assertEquals(a, "fajita")
self.assertEquals(b, "stew")
self.assertEquals(a, "fajita")
self.assertEquals(b, "stew")
self.assertEquals(c, "cake")
def test_nested_let(self):
a = "tacos"
b = "soup"
c = "cake"
with let(a="fajita"):
with let(c="bread"):
self.assertEquals(a, "fajita")
self.assertEquals(b, "soup")
self.assertEquals(c, "bread")
c = "cupcake"
self.assertEquals(c, "cupcake")
self.assertEquals(a, "fajita")
self.assertEquals(b, "soup")
self.assertEquals(c, "cake")
self.assertEquals(a, "tacos")
self.assertEquals(b, "soup")
self.assertEquals(c, "cake")
def test_local_del(self):
a = "pizza"
b = "beer"
with let(a="tacos", b="soda") as my_scope:
del a
del b
self.assertTrue("a" not in my_scope)
self.assertTrue("b" not in my_scope)
self.assertEquals(a, "pizza")
self.assertEquals(b, "beer")
def test_nonlocal(self):
with let(a="tacos", b="soda") as my_scope:
self.assertEquals(a, "tacos")
self.assertEquals(b, "soda")
self.assertTrue("a" in my_scope)
self.assertTrue("b" in my_scope)
self.assertTrue("a" not in locals())
self.assertTrue("a" not in globals())
self.assertTrue("b" not in locals())
self.assertTrue("b" not in globals())
def test_scope_in_use(self):
with let(a=1, b=2) as scope:
self.assertRaises(ScopeInUse, scope.__enter__)
# raise the exception again, but keep it around for
# observation
with scope:
try:
scope.__enter__()
except ScopeInUse as siu:
pass
# check the error proprties were set correctly
self.assertEquals(scope, siu.scope)
self.assertEquals(currentframe(), siu.frame)
def test_closure(self):
"""
This tests that closures are created over the correct scope, and
that the correct cells are made available.
"""
a = [ "tacos" ]
b = [ None ]
def get_a_1():
return a[0]
def set_a_1(new_a):
b[0] = a[0]
a[0] = new_a
def old_a_1():
return b[0]
with let(a = ["fajita"], b = [None]) as scope:
self.assertEquals(a[0], "fajita")
def get_a_2():
return a[0]
def set_a_2(new_a):
b[0] = a[0]
a[0] = new_a
def old_a_2():
return b[0]
self.assertEquals(get_a_2(), "fajita")
self.assertEquals(get_a_1(), "tacos")
self.assertEquals(old_a_1(), None)
self.assertEquals(get_a_2(), "fajita")
self.assertEquals(old_a_2(), None)
set_a_1("pizza")
set_a_2("curry")
self.assertEquals(get_a_1(), "pizza")
self.assertEquals(old_a_1(), "tacos")
self.assertEquals(get_a_2(), "curry")
self.assertEquals(old_a_2(), "fajita")
self.assertEquals(a[0], "pizza")
self.assertEquals(b[0], "tacos")
with scope:
self.assertEquals(a[0], "curry")
self.assertEquals(b[0], "fajita")
def test_accessors(self):
a = "tacos"
with let(a="pizza", b="beer") as my_scope:
pass
self.assertEquals(my_scope["a"], "pizza")
self.assertEquals(my_scope["b"], "beer")
self.assertRaises(KeyError, lambda: my_scope["c"])
my_scope["a"] = "burger"
my_scope["c"] = "fries"
self.assertEquals(my_scope["c"], "fries")
self.assertTrue("a" in my_scope)
self.assertTrue("b" in my_scope)
self.assertTrue("c" in my_scope)
with my_scope:
self.assertEquals(a, "burger")
self.assertEquals(b, "beer")
self.assertEquals(c, "fries")
del my_scope["a"]
del my_scope["c"]
self.assertRaises(KeyError, lambda: my_scope["a"])
self.assertRaises(KeyError, lambda: my_scope["c"])
self.assertTrue("a" not in my_scope)
self.assertTrue("b" in my_scope)
self.assertTrue("c" not in my_scope)
with my_scope:
self.assertEquals(a, "tacos")
self.assertEquals(b, "beer")
self.assertRaises(NameError, lambda: c)
def do_del(var):
del my_scope[var]
self.assertRaises(KeyError, do_del, "a")
self.assertRaises(KeyError, do_del, "c")
def test_with_globals(self):
self.assertEquals(_b, "soda")
_a = "pizza"
with let(_b="beer"):
self.assertEquals(_a, "pizza")
self.assertEquals(_b, "beer")
# make sure we did NOT munge _a into globals, since we
# didn't need to do so
self.assertEquals(globals()["_a"], "tacos")
# since _b was not in co_varnames, it was munged into
# globals for this scope
self.assertEquals(globals()["_b"], "beer")
# check that _b was returned to the original global val
self.assertEquals(globals()["_b"], "soda")
def test_del_global(self):
# testing `DELETE_GLOBAL` opcode. Since `b` is NOT already
# defined when we push our scope, AND since `b` is not
# ASSIGNED to anything in our block below, `b` will be
# injected as a global -- read-only from our perspective,
# because if the compiler had detected a write, it would have
# made it into fast locals. This explanation is awful, rewrite
# it.
self.assertTrue("a" not in locals())
self.assertTrue("a" not in globals())
with let(a="pizza") as scope:
self.assertEquals(a, "pizza")
del a
self.assertTrue("a" not in locals())
self.assertTrue("a" not in locals())
self.assertTrue("a" not in globals())
def test_unasigned_closure(self):
def getter():
return ourval
with let(ourval="pizza"):
# ourval is later defined, but not currently assigned, but
# we capture a closure to the unassigned cell above and again
# from within this scope.
def other_getter():
return ourval
# when the scope exits, it should be putting back an undefined
# cell value.
self.assertEquals(other_getter(), "pizza")
self.assertRaises(NameError, getter)
ourval = "tacos"
self.assertEquals(getter(), "tacos")
self.assertEquals(other_getter(), "pizza")
