def test_find_flat_nodes(self): subtree = config.E.target( config.E.path( location="A", state="evicted" ), config.E.path( location="A/B/C", state="included" ), config.E.path( location="A/B/C/D", state="evicted" ), config.E.path( location="A/E", state="included" ) ) self.assertSequenceEqual( list(config.extract_flat_nodes(subtree)), [ (target.State.EVICTED, "A"), (target.State.INCLUDED, "A/B/C"), (target.State.EVICTED, "A/B/C/D"), (target.State.INCLUDED, "A/E"), ] )
def test_embed_flat_nodes(self): flat_nodes = [ (target.State.EVICTED, "A"), (target.State.INCLUDED, "A/B/C"), (target.State.EVICTED, "A/B/C/D"), (target.State.INCLUDED, "A/E"), ] subtree = config.E.target() config.embed_flat_nodes( subtree, flat_nodes ) self.assertSequenceEqual( list(config.extract_flat_nodes(subtree)), flat_nodes )