import pytest from main import search_hashes @pytest.mark.parametrize('given, match, expected', [ ('abcdef', '00000', 609043), ('pqrstuv', '00000', 1048970) ]) def test_searching(given, match, expected): assert search_hashes(given, match) == expected