waybackpy/tests/test_utils.py

from waybackpy import __version__
from waybackpy.utils import (
    DEFAULT_USER_AGENT,
    latest_version_github,
    latest_version_pypi,
)


def test_default_user_agent():
    assert (
        DEFAULT_USER_AGENT
        == "waybackpy %s - https://github.com/akamhy/waybackpy" % __version__
    )


def test_latest_version():
    package_name = "waybackpy"
    assert latest_version_github(package_name) == latest_version_pypi(package_name)