Ejemplos de HubIO._freeze_jina_version en Python

Ejemplo n.º 1

Archivo: test_hub_build.py Proyecto: yuanl/jina

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

Ejemplo n.º 2

Archivo: test_hub_build.py Proyecto: yuanl/jina

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__}'

Ejemplo n.º 3

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()