formats: improve test coverage

8 years ago · 07cbe65875
parent 180120e787
commit 07cbe65875
1 changed files with 8 additions and 0 deletions
--- a/trezor_agent/tests/test_formats.py
+++ b/trezor_agent/tests/test_formats.py
@ -93,3 +93,11 @@ def test_curve_mismatch():
 def test_serialize_error():
    with pytest.raises(TypeError):
        formats.serialize_verifying_key(None)
+
+
+def test_get_ecdh_curve_name():
+    for c in [formats.CURVE_NIST256, formats.ECDH_CURVE25519]:
+        assert c == formats.get_ecdh_curve_name(c)
+
+    assert (formats.ECDH_CURVE25519 ==
+            formats.get_ecdh_curve_name(formats.CURVE_ED25519))