| 12345678910111213141516171819202122232425262728293031323334 |
- # Copyright (c) OpenMMLab. All rights reserved.
- from click.testing import CliRunner
- from mim.commands.install import cli as install
- from mim.commands.list import list_package
- from mim.commands.uninstall import cli as uninstall
- 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_list():
- runner = CliRunner()
- # mim install mmcls==0.23.0 --yes
- result = runner.invoke(install, ['mmcls==0.23.0', '--yes'])
- assert result.exit_code == 0
- # mim list
- target = ('mmcls', '0.23.0',
- 'https://github.com/open-mmlab/mmclassification')
- result = list_package()
- assert target in result
- def teardown_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
|
