Add tests for `_flatten_dict`.