def test_changing_y_to_none_removes_associated_yerr_columns(self):
"""
Test to check if a first column is marked as Y, a second column YErr is associated with it, but then
the first one is changed to X - the YErr mark should be removed
"""
mc = MarkedColumns()
mc.add_y(4)
ec = ErrorColumn(column=2, related_y_column=4, label_index=0)
mc.add_y_err(ec)
# check that we have both a Y col and an associated YErr
self.assertEqual(1, len(mc.as_y))
self.assertEqual(1, len(mc.as_y_err))
mc.remove(4)
# changing the column to NONE should have removed it from X, Y and YErr
self.assertEqual(0, len(mc.as_x))
self.assertEqual(0, len(mc.as_y))
self.assertEqual(0, len(mc.as_y_err))
def test_changing_y_to_none_removes_associated_yerr_columns(self):
"""
Test to check if a first column is marked as Y, a second column YErr is associated with it, but then
the first one is changed to X - the YErr mark should be removed
"""
mc = MarkedColumns()
mc.add_y(4)
ec = ErrorColumn(column=2, related_y_column=4, label_index=0)
mc.add_y_err(ec)
# check that we have both a Y col and an associated YErr
self.assertEqual(1, len(mc.as_y))
self.assertEqual(1, len(mc.as_y_err))
mc.remove(4)
# changing the column to NONE should have removed it from X, Y and YErr
self.assertEqual(0, len(mc.as_x))
self.assertEqual(0, len(mc.as_y))
self.assertEqual(0, len(mc.as_y_err))
def test_remove_column(self):
mc = MarkedColumns()
mc.add_y(4)
mc.add_x(3)
ec = ErrorColumn(column=2, related_y_column=6)
mc.add_y_err(ec)
self.assertEqual(1, len(mc.as_x))
self.assertEqual(1, len(mc.as_y))
self.assertEqual(1, len(mc.as_y_err))
mc.remove(4)
self.assertEqual(0, len(mc.as_y))
self.assertEqual(1, len(mc.as_y_err))
self.assertEqual(1, len(mc.as_x))
mc.remove(3)
self.assertEqual(0, len(mc.as_x))
self.assertEqual(0, len(mc.as_y))
self.assertEqual(1, len(mc.as_y_err))
mc.remove(2)
self.assertEqual(0, len(mc.as_x))
self.assertEqual(0, len(mc.as_y))
self.assertEqual(0, len(mc.as_y_err))
def test_remove_column(self):
mc = MarkedColumns()
mc.add_y(4)
mc.add_x(3)
ec = ErrorColumn(column=2, related_y_column=6, label_index=0)
mc.add_y_err(ec)
self.assertEqual(1, len(mc.as_x))
self.assertEqual(1, len(mc.as_y))
self.assertEqual(1, len(mc.as_y_err))
mc.remove(4)
self.assertEqual(0, len(mc.as_y))
self.assertEqual(1, len(mc.as_y_err))
self.assertEqual(1, len(mc.as_x))
mc.remove(3)
self.assertEqual(0, len(mc.as_x))
self.assertEqual(0, len(mc.as_y))
self.assertEqual(1, len(mc.as_y_err))
mc.remove(2)
self.assertEqual(0, len(mc.as_x))
self.assertEqual(0, len(mc.as_y))
self.assertEqual(0, len(mc.as_y_err))