From 8d65f23980693db4bcb072c870a7a9e273b3ad78 Mon Sep 17 00:00:00 2001 From: Chris Wenneman Date: Fri, 6 Oct 2023 13:34:22 -0700 Subject: [PATCH] Fixed failing unit test by removing refernce to Ensemble._nobs_band_cols field --- tests/tape_tests/test_ensemble.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/tests/tape_tests/test_ensemble.py b/tests/tape_tests/test_ensemble.py index 2b3c9bb9..e97e4892 100644 --- a/tests/tape_tests/test_ensemble.py +++ b/tests/tape_tests/test_ensemble.py @@ -299,9 +299,8 @@ def test_read_source_dict(dask_client): assert src_table.iloc[i][ens._err_col] == rows[ens._err_col][i] # Check that the derived object table is correct. - assert obj_table.shape[0] == 2 - assert obj_table.iloc[0][ens._nobs_tot_col] == 4 - assert obj_table.iloc[1][ens._nobs_tot_col] == 5 + assert 8001 in obj_table.index + assert 8002 in obj_table.index def test_insert(parquet_ensemble):