def test_jina_version_freeze_no_jina_dependency(requirements, tmpdir): import pkg_resources args = set_hub_build_parser().parse_args([str(tmpdir)]) hubio = HubIO(args) hubio._freeze_jina_version() requirements_file = os.path.join(tmpdir, 'requirements.txt') with open(requirements_file, 'r') as fp: requirements = pkg_resources.parse_requirements(fp) assert len(list(filter(lambda x: 'jina' in str(x), requirements))) == 0
def test_jina_version_freeze(requirements, tmpdir): import pkg_resources from jina import __version__ args = set_hub_build_parser().parse_args([str(tmpdir)]) hubio = HubIO(args) hubio._freeze_jina_version() requirements_file = os.path.join(tmpdir, 'requirements.txt') with open(requirements_file, 'r') as fp: requirements = pkg_resources.parse_requirements(fp) assert len(list(filter(lambda x: 'jina' in str(x), requirements))) == 1 for req in requirements: if 'jina' in str(req): assert str(req) == f'jina=={__version__}'
def test_jina_version_freeze_no_jina_dependency_git_no_raise(requirements, tmpdir): args = set_hub_build_parser().parse_args([str(tmpdir)]) hubio = HubIO(args) hubio._freeze_jina_version()