Assert metadata are equal.
(got, exp)
| 4171 | |
| 4172 | |
| 4173 | def assert_metadata_equal(got, exp): |
| 4174 | """Assert metadata are equal.""" |
| 4175 | if exp is None: |
| 4176 | assert got is None |
| 4177 | elif isinstance(exp, list): |
| 4178 | assert isinstance(got, list) |
| 4179 | assert len(got) == len(exp) |
| 4180 | for ii, (g, e) in enumerate(zip(got, exp)): |
| 4181 | assert list(g.keys()) == list(e.keys()) |
| 4182 | for key in g.keys(): |
| 4183 | assert g[key] == e[key], (ii, key) |
| 4184 | else: # DataFrame |
| 4185 | import pandas |
| 4186 | |
| 4187 | assert isinstance(exp, pandas.DataFrame) |
| 4188 | assert isinstance(got, pandas.DataFrame) |
| 4189 | assert set(got.columns) == set(exp.columns) |
| 4190 | check = got == exp |
| 4191 | assert check.all().all() |
| 4192 | |
| 4193 | |
| 4194 | @pytest.mark.parametrize( |
no test coverage detected