test: severity_from_level

2022-02-10 12:20:42 +01:00
parent b1ce2f290b
commit 32de447205
2 changed files with 27 additions and 6 deletions
--- a/tests/test_format.py
+++ b/tests/test_format.py
@@ -22,3 +22,19 @@ class TextToXMLTestCase(unittest.TestCase):
    def test_specials_chars(self):
        for inp, out in self.specials.items():
            self.assertEqual(escape_xml(inp), out)
+
+
+class CodeClimateSeverityTestCase(unittest.TestCase):
+
+    expected = {
+        None: "info",
+        'warning': "minor",
+        'error': "major",
+        0: "info",
+        1: "minor",
+        2: "major",
+    }
+
+    def test_specials_chars(self):
+        for inp, out in self.expected.items():
+            self.assertEqual(severity_from_level(inp), out)