mirror of https://github.com/open-mmlab/mim.git
52 lines
1.7 KiB
Python
52 lines
1.7 KiB
Python
from click.testing import CliRunner
|
|
|
|
from mim.commands.install import cli as install
|
|
from mim.commands.uninstall import cli as uninstall
|
|
from mim.utils import read_installation_records
|
|
|
|
|
|
def setup_module():
|
|
runner = CliRunner()
|
|
result = runner.invoke(uninstall, ['mmcv-full', '--yes'])
|
|
assert result.exit_code == 0
|
|
result = runner.invoke(uninstall, ['mmcls', '--yes'])
|
|
assert result.exit_code == 0
|
|
|
|
|
|
def test_read_installation_records():
|
|
target = ('mmcls', '0.11.0',
|
|
'https://github.com/open-mmlab/mmclassification.git')
|
|
pkgs_info = read_installation_records()
|
|
assert target not in pkgs_info
|
|
|
|
runner = CliRunner()
|
|
result = runner.invoke(install, ['mmcls==0.11.0', '--yes'])
|
|
assert result.exit_code == 0
|
|
target = ('mmcls', '0.11.0',
|
|
'https://github.com/open-mmlab/mmclassification.git')
|
|
pkgs_info = read_installation_records()
|
|
assert target in pkgs_info
|
|
|
|
result = runner.invoke(uninstall, ['mmcls', '--yes'])
|
|
assert result.exit_code == 0
|
|
|
|
|
|
def test_write_installation_records():
|
|
runner = CliRunner()
|
|
result = runner.invoke(install, ['mmcls==0.11.0', '--yes'])
|
|
assert result.exit_code == 0
|
|
target = ('mmcls', '0.11.0',
|
|
'https://github.com/open-mmlab/mmclassification.git')
|
|
pkgs_info = read_installation_records()
|
|
assert target in pkgs_info
|
|
|
|
result = runner.invoke(install, ['mmcls==0.10.0', '--yes'])
|
|
assert result.exit_code == 0
|
|
new_target = ('mmcls', '0.10.0',
|
|
'https://github.com/open-mmlab/mmclassification.git')
|
|
pkgs_info = read_installation_records()
|
|
assert target not in pkgs_info and new_target in pkgs_info
|
|
|
|
result = runner.invoke(uninstall, ['mmcls', '--yes'])
|
|
assert result.exit_code == 0
|