def test_checker_trained_model_1(setup):
checker = Checker(setup['name'])
try:
checker.trained_model = None
except TypeError:
assert True
else:
assert False, 'should got error'
def test_checker_from_cp(setup):
checker = Checker(setup['name'])
name = checker.name
path = checker.path
checker(**setup['cp'])
checker2 = Checker.load(path + '/' + name)
model_state, other = checker2[0]
other_ = deepcopy(setup['cp'])
del other_['model_state']
assert model_state == setup['cp']['model_state']
assert other == other_
def test_checker_init_model_3(setup):
checker = Checker(setup['name'])
checker.init_model = setup['model']
assert str(checker.init_model) == str(setup['model'])
def test_checker_init_model_2(setup):
checker = Checker(setup['name'])
try:
checker.init_model = setup['model']
except TypeError:
assert False, 'should not got error'
def test_checker_assign_path(setup):
checker = Checker(setup['name'], path=setup['dot'])
assert isinstance(checker, LocalStorage)
assert checker.path == str(Path().resolve())
assert checker.name == setup['name'] + '@1'
def test_checker_default_path(setup):
checker = Checker(setup['name'], path=setup['default'])
assert isinstance(checker, LocalStorage)
assert checker.path == setup['default']
assert checker.name == setup['name'] + '@2'
def test_checker_call(setup):
checker = Checker(setup['name'])
checker(**setup['cp'])
assert (Path(checker.path) / checker.name / 'checkpoints').exists()
def test_checker_trained_model_3(setup):
checker = Checker(setup['name'])
checker.trained_model = setup['model']
assert str(checker.trained_model) == str(setup['model'])
def test_checker_trained_model_2(setup):
checker = Checker(setup['name'])
checker.trained_model = setup['model']
def test_checker_trained_model_1(setup):
checker = Checker(setup['name'])
with pytest.raises(TypeError):
checker.trained_model = None
def test_checker_init_model_2(setup):
checker = Checker(setup['name'])
checker.init_model = setup['model']
def test_checker_omit_path(setup):
checker = Checker(setup['name'])
assert isinstance(checker, Storage)
assert checker.path == setup['default']
assert checker.name == setup['name'] + '@1'
