def test_quick_fixes(self):
err = PromptingError([
PromptingError.WrongColumnType(["A"], "text",
frozenset({"number"})),
PromptingError.WrongColumnType(["B", "C"], "datetime",
frozenset({"number"})),
])
quick_fixes_result = err.as_quick_fixes()
self.assertEqual(
quick_fixes_result,
[
QuickFix(
"Convert Text to Numbers",
"prependModule",
["converttexttonumber", {
"colnames": ["A"]
}],
),
QuickFix(
"Convert Dates & Times to Numbers",
"prependModule",
["converttexttonumber", {
"colnames": ["B", "C"]
}],
),
],
)
error_result = err.as_error_str()
self.assertEqual(
error_result,
("The column “A” must be converted from Text to Numbers.\n\n"
"The columns “B” and “C” must be converted from Dates & Times to Numbers."
),
)
def test_as_quick_fixes(self):
err = PromptingError([
PromptingError.WrongColumnType(["A"], "text",
frozenset({"number"})),
PromptingError.WrongColumnType(["B", "C"], "datetime",
frozenset({"number"})),
])
result = err.as_quick_fixes()
self.assertEqual(
result,
[
QuickFix(
'Convert "A" to Numbers',
"prependModule",
["converttexttonumber", {
"colnames": ["A"]
}],
),
QuickFix(
'Convert "B", "C" to Numbers',
"prependModule",
["converttexttonumber", {
"colnames": ["B", "C"]
}],
),
],
)
def test_as_quick_fixes(self):
err = PromptingError([
PromptingError.WrongColumnType(['A'], 'text',
frozenset({'number'})),
PromptingError.WrongColumnType(['B', 'C'], 'datetime',
frozenset({'number'})),
])
result = err.as_quick_fixes()
self.assertEqual(result, [
QuickFix('Convert "A" to Numbers', 'prependModule',
['converttexttonumber', {
'colnames': ['A']
}]),
QuickFix('Convert "B", "C" to Numbers', 'prependModule',
['converttexttonumber', {
'colnames': ['B', 'C']
}]),
])
def test_quick_fixes_convert_to_text(self):
err = PromptingError([
PromptingError.WrongColumnType(["A", "B"], None,
frozenset({"text"}))
])
quick_fixes_result = err.as_quick_fixes()
self.assertEqual(
quick_fixes_result,
[
QuickFix(
I18nMessage.TODO_i18n("Convert to Text"),
QuickFixAction.PrependStep("converttotext",
{"colnames": ["A", "B"]}),
)
],
)
error_result = err.as_error_str()
self.assertEqual(error_result,
"The columns “A” and “B” must be converted to Text.")
def test_quick_fixes_convert_to_text(self):
err = PromptingError([
PromptingError.WrongColumnType(["A", "B"], None,
frozenset({"text"}))
])
quick_fixes_result = err.as_quick_fixes()
self.assertEqual(
quick_fixes_result,
[
QuickFix(
"Convert to Text",
"prependModule",
["converttotext", {
"colnames": ["A", "B"]
}],
)
],
)
error_result = err.as_error_str()
self.assertEqual(error_result,
"The columns “A” and “B” must be converted to Text.")
