def test_unpack_keys_basic(self):
obj = SimpleNamespace(thing=SimpleNamespace())
d = dict(foo=1, bar=3, spam=7, baz=8, x=9)
out = {}
foo, obj.thing.spam, obj.bar, out['baz'] = unpack_keys(d)
self.assertEqual(foo, d['foo'])
self.assertEqual(obj.bar, d['bar'])
self.assertEqual(obj.thing.spam, d['spam'])
self.assertEqual(out, {'baz': d['baz']})
def test_unpack_attrs(self):
obj = SimpleNamespace(aa='bv', bb='cc', cc='aa')
cc, bb, aa = unpack_attrs(obj)
self.assertEqual(aa, obj.aa)
self.assertEqual(bb, obj.bb)
self.assertEqual(cc, obj.cc)
d, e = unpack_attrs(obj, default=9)
assert d == e == 9
def test_ipython_cell_code():
assert is_ipython_cell_code(
SimpleNamespace(
co_name="<cell line: 1>",
co_filename="tmp/ipykernel_3/foo",
))
assert not is_ipython_cell_code(
SimpleNamespace(
co_name="<cell line: 1",
co_filename="tmp/ipykernel_3/foo",
))
assert not is_ipython_cell_code(
SimpleNamespace(
co_name="<cell line: 1>",
co_filename="tmp/ipykernel_3",
))
def test_dict_of(self):
a = 1
obj = SimpleNamespace(b=2)
self.assertEqual(spells.dict_of(
a, obj.b,
c=3, d=4
), dict(
a=a, b=obj.b,
c=3, d=4))