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