diff --git a/tests/test_password.py b/tests/test_password.py index 2962859..35ddf08 100644 --- a/tests/test_password.py +++ b/tests/test_password.py @@ -15,3 +15,9 @@ def test_no_digits(): def test_invalid_length(): with pytest.raises(ValueError): generate(0) + +def test_symbols_gen(): + assert any(c in string.punctuation for c in generate(200, use_symbols=True)) + +def test_no_symbols_gen(): + assert not any(c in string.punctuation for c in generate(200))