Ejemplo n.º 1
0
 def test_assert_empty(self):
     config_dict = {"test": "hello"}
     params = Params(config_dict)
     with pytest.raises(ConfigurationError):
         params.assert_empty("dummy")
     assert params.pop("test") == "hello"
     params.assert_empty("dummy")
Ejemplo n.º 2
0
    def test_write_read_from_file(self):
        config_dict = {"test": "hello"}
        params = Params(config_dict)
        assert params.as_dict() == config_dict

        write_path = self.TEMPORARY_DIR / "dummy_config.json"
        params.to_file(str(write_path))

        params2 = Params.from_file(str(write_path))

        assert params.as_dict() == params2.as_dict()

        assert params.pop("test") == "hello"
        assert params.pop("test2", "none") == "none"
        with pytest.raises(ConfigurationError):
            params.pop("test")
Ejemplo n.º 3
0
    def test_pop_nested_param(self):
        config_dict = {"model": {"type": "test", "other_param": 1}}

        params = Params(config_dict)

        assert isinstance(params.pop("model"), Params)