Instead of a separate two line method it would be clearer to just assign this dict to `expected_output` and do the loop here.