Use OmegaConf.set_readonly() in tests

omry · Apr 16, 2021 · ac26495 · ac26495
1 parent c372a58
commit ac26495
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/interpolation/built_in_resolvers/test_dict.py b/tests/interpolation/built_in_resolvers/test_dict.py
@@ -274,7 +274,7 @@ def test_dict_values_are_typed() -> None:
 )
 def test_readonly_parent(cfg: Any, expected: Any) -> None:
     cfg = OmegaConf.create(cfg)
-    cfg._set_flag("readonly", True)
+    OmegaConf.set_readonly(cfg, True)
     assert cfg.x == expected
 
 

diff --git a/tests/interpolation/test_custom_resolvers.py b/tests/interpolation/test_custom_resolvers.py
@@ -351,7 +351,7 @@ def test_resolver_output_dict(restore_resolvers: Any, readonly: bool) -> None:
     some_dict = {"a": 0, "b": "${y}"}
     OmegaConf.register_new_resolver("dict", lambda: some_dict)
     c = OmegaConf.create({"x": "${dict:}", "y": -1})
-    c._set_flag("readonly", readonly)
+    OmegaConf.set_readonly(c, readonly)
     assert c.x == some_dict
     x_node = dereference_node(c, "x")
     assert isinstance(x_node, AnyNode)
@@ -363,7 +363,7 @@ def test_resolver_output_list(restore_resolvers: Any, readonly: bool) -> None:
     some_list = ["a", 0, "${y}"]
     OmegaConf.register_new_resolver("list", lambda: some_list)
     c = OmegaConf.create({"x": "${list:}", "y": -1})
-    c._set_flag("readonly", readonly)
+    OmegaConf.set_readonly(c, readonly)
     assert c.x == some_list
     x_node = dereference_node(c, "x")
     assert isinstance(x_node, AnyNode)