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.")