def test_expand_pages(): pages = {k: pageDict[k] for k in ('F')} pagesTwo = {k: pageDict[k] for k in ('A', 'E')} expanded_pages = expand_pages(pages) assert all(x in expanded_pages for x in ['F', 'E']) assert all(x not in expanded_pages for x in ['A', 'B', 'C', 'D']) expanded_pages = expand_pages(pagesTwo) print(expanded_pages) assert all(x in expanded_pages for x in ['A', 'B', 'C', 'D', 'E', 'F'])