timeindex vs numindex fix

saqc/core/core.py

@@ -145,7 +145,7 @@ def runner(metafname, flagger, data, flags=None, nodata=np.nan):
                 plot(dchunk, ffchunk, mask, varname, flagger, title=flag_test)
 
             data.loc[start_date:end_date] = dchunk
-            flags[start_date:end_date] = ffchunk.squeeze()
+            flags.loc[start_date:end_date] = ffchunk.squeeze()
 
         flagger.nextTest()
 

test/test_core.py

@@ -41,7 +41,7 @@ def test_positionalPartitioning(flagger):
         vname, start_index, end_index = row[fields]
         fchunk = pflags.loc[flagger.isFlagged(pflags[vname]), vname]
         assert fchunk.index.min() == start_index, "different start indices"
-        assert fchunk.index.max() + 1 == end_index, f"different end indices: {fchunk.index.max()} vs. {end_index}"
+        assert fchunk.index.max() == end_index, f"different end indices: {fchunk.index.max()} vs. {end_index}"
 
 
 @pytest.mark.parametrize("flagger", TESTFLAGGERS)