def test_parse_custom_list_ignore_trailing_newline(self):
self.assertEqual(
_parse_custom_list("X\nY\n", ["A", "B"]),
{
"A": "X",
"B": "Y"
}, # no ValueError
)
def test_parse_custom_list_newline_means_ignore_commas(self):
self.assertEqual(
_parse_custom_list("X,Y\nZ,A\nB,C", ["A", "B", "C"]),
{
"A": "X,Y",
"B": "Z,A",
"C": "B,C"
},
)
def test_parse_custom_list_by_comma(self):
self.assertEqual(
_parse_custom_list("X, Y, Z", ["A", "B", "C"]),
{
"A": "X",
"B": "Y",
"C": "Z"
},
)
def test_parse_custom_list_by_newline(self):
self.assertEqual(
_parse_custom_list("X\nY\nZ", ["A", "B", "C"]),
{
"A": "X",
"B": "Y",
"C": "Z"
},
)
def test_parse_custom_list_trailing_newline_still_split_by_comma(self):
"""If the user added a newline to the end, it's still commas."""
self.assertEqual(
_parse_custom_list("X, Y, Z\n", ["A", "B", "C"]),
{
"A": "X",
"B": "Y",
"C": "Z"
},
)
def test_parse_custom_list_skip_whitespace_columns(self):
self.assertEqual(_parse_custom_list("X\n\nZ", ["A", "B", "C"]), {
"A": "X",
"C": "Z"
})
def test_parse_custom_list_too_many_columns_is_valueerror(self):
with self.assertRaisesRegex(
ValueError,
"You supplied 4 column names, but the table has 3 columns."):
_parse_custom_list("A\nB\nC\nD", ["A", "B", "C"])
def test_parse_custom_list_ignore_no_op_renames(self):
self.assertEqual(_parse_custom_list("A\nY\nC", ["A", "B", "C"]),
{"B": "Y"})
def test_parse_custom_list_allow_too_few_columns(self):
self.assertEqual(_parse_custom_list("X\nY", ["A", "B", "C"]), {
"A": "X",
"B": "Y"
})