forked from langroid/langroid
-
Notifications
You must be signed in to change notification settings - Fork 0
/
test_global_state.py
47 lines (33 loc) · 1.35 KB
/
test_global_state.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
from langroid.utils.globals import GlobalState
class _TestGlobals(GlobalState):
"""Test-specific global variables.
(This is how users should define their own global variables)
"""
some_variable: int = 0
another_variable: str = ""
def test_initial_global_state():
"""
Test that the global state initializes with the default values.
"""
assert _TestGlobals.get_value("some_variable") == 0
assert _TestGlobals.get_value("another_variable") == ""
def test_set_global_state():
"""
Test setting new values on the global state.
"""
_TestGlobals.set_values(some_variable=5, another_variable="Test")
assert _TestGlobals.get_value("some_variable") == 5
assert _TestGlobals.get_value("another_variable") == "Test"
_TestGlobals.set_values(some_variable=7, another_variable="hello")
assert _TestGlobals.get_value("some_variable") == 7
assert _TestGlobals.get_value("another_variable") == "hello"
def test_singleton_behavior():
"""
Test that the global state behaves as a singleton.
"""
first_instance = _TestGlobals.get_instance()
second_instance = _TestGlobals.get_instance()
assert first_instance is second_instance
# Modify using one instance and check with the other
first_instance.set_values(some_variable=10)
assert second_instance.get_value("some_variable") == 10