def test_setdefault(): omd = OMD() empty_list = [] x = omd.setdefault('1', empty_list) assert x is empty_list y = omd.setdefault('2') assert y is None assert omd.setdefault('1', None) is empty_list e_omd = OMD() e_omd.addlist(1, []) assert e_omd.popall(1, None) is None assert len(e_omd) == 0
def test_pop_all(): omd = OMD() omd.add('even', 0) omd.add('odd', 1) omd.add('even', 2) assert omd.popall('odd') == [1] assert len(omd) == 1 try: omd.popall('odd') assert False except KeyError: pass assert omd.popall('odd', None) is None assert omd.popall('even') == [0, 2] assert len(omd) == 0 assert omd.popall('nope', None) is None
def test_pop_all(): omd = OMD() omd.add("even", 0) omd.add("odd", 1) omd.add("even", 2) assert omd.popall("odd") == [1] assert len(omd) == 1 try: omd.popall("odd") assert False except KeyError: pass assert omd.popall("odd", None) is None assert omd.popall("even") == [0, 2] assert len(omd) == 0 assert omd.popall("nope", None) is None