def test_load_invalid_target(self, tmpdir): directory = tmpdir.mkdir('ckpt') with raises(FileNotFoundError): Population.load(directory.join('no_file' + self.extension), lambda x: x) with raises(FileNotFoundError): Population.load(directory, lambda x: x) txt_file = directory.join('file.txt') txt_file.write('Something') with raises(ValueError): Population.load(txt_file.strpath, lambda x: x)
def test_load(self, tmpdir, simple_population): directory = tmpdir.mkdir("ckpt") simple_population.checkpoint(target=directory, method=self.method) pop = Population.load(directory, lambda x: x['x']) assert len(simple_population) == len(pop) assert all(x.__dict__ == y.__dict__ for x, y in zip(simple_population, pop))