Fixed failing unit test by removing refernce to

Ensemble._nobs_band_cols field
lincc-frameworks · Oct 6, 2023 · 8d65f23 · 8d65f23
1 parent 0812e86
commit 8d65f23
Showing 1 changed file with 2 additions and 3 deletions.
diff --git 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):