add some missing type hints to test state vars

This commit is contained in:
Benedikt Bartscher 2024-02-29 22:52:10 +01:00
parent 4380cd266d
commit 80f02753ba
No known key found for this signature in database

View File

@ -10,7 +10,7 @@ class DictMutationTestState(BaseState):
"""A state for testing ReflexDict mutation."""
# plain dict
details = {"name": "Tommy"}
details: Dict[str, str] = {"name": "Tommy"}
def add_age(self):
"""Add an age to the dict."""
@ -37,7 +37,7 @@ class DictMutationTestState(BaseState):
self.details.pop("age")
# dict in list
address = [{"home": "home address"}, {"work": "work address"}]
address: List[Dict[str,str]] = [{"home": "home address"}, {"work": "work address"}]
def remove_home_address(self):
"""Remove the home address from dict in the list."""
@ -48,7 +48,7 @@ class DictMutationTestState(BaseState):
self.address[0]["street"] = "street address"
# nested dict
friend_in_nested_dict = {"name": "Nikhil", "friend": {"name": "Alek"}}
friend_in_nested_dict: Dict[str, Union[str, Dict[str,str]]] = {"name": "Nikhil", "friend": {"name": "Alek"}}
def change_friend_name(self):
"""Change the friend's name in the nested dict."""
@ -67,7 +67,7 @@ class ListMutationTestState(BaseState):
"""A state for testing ReflexList mutation."""
# plain list
plain_friends = ["Tommy"]
plain_friends: List[str] = ["Tommy"]
def make_friend(self):
"""Add a friend to the list."""
@ -99,7 +99,7 @@ class ListMutationTestState(BaseState):
self.plain_friends.remove("Tommy")
# list in dict
friends_in_dict = {"Tommy": ["Jenny"]}
friends_in_dict: Dict[str, List[str]] = {"Tommy": ["Jenny"]}
def remove_jenny_from_tommy(self):
"""Remove Jenny from Tommy's friends list."""
@ -114,7 +114,7 @@ class ListMutationTestState(BaseState):
self.friends_in_dict["Tommy"].clear()
# nested list
friends_in_nested_list = [["Tommy"], ["Jenny"]]
friends_in_nested_list: List[List[str]] = [["Tommy"], ["Jenny"]]
def remove_first_group(self):
"""Remove the first group of friends from the nested list."""