Add additional tests to select and checkbox (#131)

- Test when a `Choice` instance is passed as `default`
   or `initial_choice`.

 - Test when a `Choice` instance and string have the
   same value and `default` or `initial_choice` is
   set to the `Choice` instance.

tests/prompts/test_checkbox.py

@@ -185,6 +185,16 @@ def test_list_ctr_c():
 
 
 def test_checkbox_initial_choice():
+    message = "Foo message"
+    choice = Choice("bazz")
+    kwargs = {"choices": ["foo", choice], "initial_choice": choice}
+    text = KeyInputs.SPACE + KeyInputs.ENTER + "\r"
+
+    result, cli = feed_cli_with_input("checkbox", message, text, **kwargs)
+    assert result == ["bazz"]
+
+
+def test_select_initial_choice_string():
     message = "Foo message"
     kwargs = {"choices": ["foo", "bazz"], "initial_choice": "bazz"}
     text = KeyInputs.SPACE + KeyInputs.ENTER + "\r"
@@ -193,6 +203,16 @@ def test_checkbox_initial_choice():
     assert result == ["bazz"]
 
 
+def test_select_initial_choice_duplicate():
+    message = "Foo message"
+    choice = Choice("foo")
+    kwargs = {"choices": ["foo", choice, "bazz"], "initial_choice": choice}
+    text = KeyInputs.DOWN + KeyInputs.SPACE + KeyInputs.ENTER + "\r"
+
+    result, cli = feed_cli_with_input("checkbox", message, text, **kwargs)
+    assert result == ["bazz"]
+
+
 def test_checkbox_initial_choice_not_selectable():
     message = "Foo message"
     separator = Separator()

tests/prompts/test_select.py

@@ -173,6 +173,16 @@ def test_allow_shortcut_key_with_True():
 
 
 def test_select_initial_choice():
+    message = "Foo message"
+    choice = Choice("bazz")
+    kwargs = {"choices": ["foo", choice], "default": choice}
+    text = KeyInputs.ENTER + "\r"
+
+    result, cli = feed_cli_with_input("select", message, text, **kwargs)
+    assert result == "bazz"
+
+
+def test_select_initial_choice_string():
     message = "Foo message"
     kwargs = {"choices": ["foo", "bazz"], "default": "bazz"}
     text = KeyInputs.ENTER + "\r"
@@ -181,6 +191,16 @@ def test_select_initial_choice():
     assert result == "bazz"
 
 
+def test_select_initial_choice_duplicate():
+    message = "Foo message"
+    choice = Choice("foo")
+    kwargs = {"choices": ["foo", choice, "bazz"], "default": choice}
+    text = KeyInputs.DOWN + KeyInputs.ENTER + "\r"
+
+    result, cli = feed_cli_with_input("select", message, text, **kwargs)
+    assert result == "bazz"
+
+
 def test_select_initial_choice_not_selectable():
     message = "Foo message"
     separator = Separator()