diff --git a/tests/ms2_filler/test_forest.py b/tests/ms2_filler/test_forest.py index 77a80e3..6218266 100644 --- a/tests/ms2_filler/test_forest.py +++ b/tests/ms2_filler/test_forest.py @@ -61,13 +61,14 @@ def test_forest_ms2_structure(msfile): position = tb.getcell("POSITION", i) assert position.shape == (3,) assert np.allclose(position, np.array([-3871023.46, 3428106.87, 3724039.47])) + assert tb.getcell("FLAG_ROW", i) is False with open_table(os.path.join(msfile, "DATA_DESCRIPTION")) as tb: assert tb.nrows() == num_spws for i in range(num_spws): assert tb.getcell("SPECTRAL_WINDOW_ID", i) == i assert tb.getcell("POLARIZATION_ID", i) == 0 - assert tb.getcell("FLAG_ROW", i) == False + assert tb.getcell("FLAG_ROW", i) is False with open_table(os.path.join(msfile, "STATE")) as tb: intents_map = dict((i, v) for i, v in enumerate(tb.getcol("OBS_MODE"))) diff --git a/tests/ms2_filler/test_h40.py b/tests/ms2_filler/test_h40.py index 3e51171..e4ada3c 100644 --- a/tests/ms2_filler/test_h40.py +++ b/tests/ms2_filler/test_h40.py @@ -63,13 +63,14 @@ def test_h40_ms2_structure(msfile): position = tb.getcell("POSITION", i) assert position.shape == (3,) assert np.allclose(position, np.array([-3871023.46, 3428106.87, 3724039.47])) + assert tb.getcell("FLAG_ROW", i) is False with open_table(os.path.join(msfile, "DATA_DESCRIPTION")) as tb: assert tb.nrows() == num_spws for i in range(num_spws): assert tb.getcell("SPECTRAL_WINDOW_ID", i) == i assert tb.getcell("POLARIZATION_ID", i) == 0 - assert tb.getcell("FLAG_ROW", i) == False + assert tb.getcell("FLAG_ROW", i) is False with open_table(os.path.join(msfile, "STATE")) as tb: intents_map = dict((i, v) for i, v in enumerate(tb.getcol("OBS_MODE"))) diff --git a/tests/ms2_filler/test_z45.py b/tests/ms2_filler/test_z45.py index e608253..6f15e8d 100644 --- a/tests/ms2_filler/test_z45.py +++ b/tests/ms2_filler/test_z45.py @@ -62,13 +62,14 @@ def test_z45_ms2_structure(msfile): position = tb.getcell("POSITION", i) assert position.shape == (3,) assert np.allclose(position, np.array([-3871023.46, 3428106.87, 3724039.47])) + assert tb.getcell("FLAG_ROW", i) is False with open_table(os.path.join(msfile, "DATA_DESCRIPTION")) as tb: assert tb.nrows() == num_spws for i in range(num_spws): assert tb.getcell("SPECTRAL_WINDOW_ID", i) == i assert tb.getcell("POLARIZATION_ID", i) == 0 - assert tb.getcell("FLAG_ROW", i) == False + assert tb.getcell("FLAG_ROW", i) is False with open_table(os.path.join(msfile, "STATE")) as tb: intents_map = dict((i, v) for i, v in enumerate(tb.getcol("OBS_MODE")))