LightMirrors/test/mirrors_test.py
2024-06-15 12:53:19 +08:00

33 lines
1002 B
Python

import unittest
from utils import call
PYPI_HOST = "pypi.local.homeinfra.org"
PYPI_INDEX = f"http://{PYPI_HOST}/simple"
TORCH_HOST = "torch.local.homeinfra.org"
TORCH_INDEX = f"http://{TORCH_HOST}/whl"
class TestPypi(unittest.TestCase):
def test_pypi_http(self):
call(f"pip download -i {PYPI_INDEX} django --trusted-host {PYPI_HOST} --dest /tmp/pypi/")
def test_torch_http(self):
call(f"pip download -i {TORCH_INDEX} tqdm --trusted-host {TORCH_HOST} --dest /tmp/torch/")
def test_dockerhub_pull(self):
call(f"docker pull docker.local.homeinfra.org/alpine:3.12")
def test_ghcr_pull(self):
call(f"docker pull ghcr.local.homeinfra.org/linuxcontainers/alpine")
def test_quay_pull(self):
call(f"docker pull quay.local.homeinfra.org/quay/busybox")
def test_k8s_pull(self):
call(f"docker pull k8s.local.homeinfra.org/pause:3.5")
def test_nvcr_pull(self):
call(f"docker pull nvcr.local.homeinfra.org/nvidia/cuda")