def test_pop_string(): f = PlainTextFormatter() type_str = '%s.%s' % (C.__module__, 'C') with nt.assert_raises(KeyError): f.pop(type_str) f.for_type(type_str, foo_printer) f.pop(type_str) with nt.assert_raises(KeyError): f.lookup_by_type(C) with nt.assert_raises(KeyError): f.pop(type_str) f.for_type(C, foo_printer) nt.assert_is(f.pop(type_str, None), foo_printer) with nt.assert_raises(KeyError): f.lookup_by_type(C) with nt.assert_raises(KeyError): f.pop(type_str) nt.assert_is(f.pop(type_str, None), None)
def test_pop(): f = PlainTextFormatter() f.for_type(C, foo_printer) nt.assert_is(f.lookup_by_type(C), foo_printer) nt.assert_is(f.pop(C, None), foo_printer) f.for_type(C, foo_printer) nt.assert_is(f.pop(C), foo_printer) with nt.assert_raises(KeyError): f.lookup_by_type(C) with nt.assert_raises(KeyError): f.pop(C) with nt.assert_raises(KeyError): f.pop(A) nt.assert_is(f.pop(A, None), None)