def test_sanitise_label(self):
for label_in, label_out in (('X', 'X'), ('X 1', 'X 1'), ('X @', 'X '),
('X@', 'X'), ('X@@@', 'X'), ('HEX', 'HEX'),
('HE@X', 'HE@X'), ('HE@X 1', 'HE@X 1'),
('@', None), ('@X', None), ('@HEX', None)):
self.assertEqual(SBSPlotter._sanitise_label(label_in), label_out)
def test_sanitise_label(self):
for label_in, label_out in (
("X", "X"),
("X 1", "X 1"),
("X @", "X "),
("X@", "X"),
("X@@@", "X"),
("HEX", "HEX"),
("HE@X", "HE@X"),
("HE@X 1", "HE@X 1"),
("@", None),
("@X", None),
("@HEX", None),
):
self.assertEqual(SBSPlotter._sanitise_label(label_in), label_out)