From 38b2cf6c654eca9add5afe9e9ecd8174f0a0c10c Mon Sep 17 00:00:00 2001
From: Joachim Jablon <ewjoachim@gmail.com>
Date: Sat, 24 Jul 2021 00:12:23 +0200
Subject: [PATCH] TMP

---
 tests/code.py      | 3 +--
 tests/test_code.py | 8 +++++++-
 2 files changed, 8 insertions(+), 3 deletions(-)

diff --git a/tests/code.py b/tests/code.py
index 972a635..dde1ac7 100644
--- a/tests/code.py
+++ b/tests/code.py
@@ -6,5 +6,4 @@ def code(arg: Optional[bool]) -> str:
         return "a"
     elif arg is True:
         return "b"
-    else:
-        return "c"
+    return "c"
diff --git a/tests/test_code.py b/tests/test_code.py
index d3e4c7d..07a7d56 100644
--- a/tests/test_code.py
+++ b/tests/test_code.py
@@ -3,6 +3,12 @@ import pytest
 from . import code
 
 
-@pytest.mark.parametrize("arg, expected", [(None, "a")])
+@pytest.mark.parametrize(
+    "arg, expected",
+    [
+        (None, "a"),
+        (True, "b"),
+    ],
+)
 def test_code(arg, expected):
     assert code.code(arg) == expected