Use Collection from collections.abc instead of typing.List.

features/communication.py

@@ -1,4 +1,4 @@
-from typing import List
+from collections.abc import Collection
 
 import pandas as pd
 
@@ -9,7 +9,7 @@ call_types = {1: "incoming", 2: "outgoing", 3: "missed"}
 sms_types = {1: "received", 2: "sent"}
 
 
-def get_call_data(usernames: List) -> pd.DataFrame:
+def get_call_data(usernames: Collection) -> pd.DataFrame:
     """
     Read the data from the calls table and return it in a dataframe.
 
@@ -33,7 +33,7 @@ def get_call_data(usernames: List) -> pd.DataFrame:
     return df_calls
 
 
-def get_sms_data(usernames: List) -> pd.DataFrame:
+def get_sms_data(usernames: Collection) -> pd.DataFrame:
     """
     Read the data from the sms table and return it in a dataframe.
 

features/screen.py

@@ -1,4 +1,4 @@
-from typing import List
+from collections.abc import Collection
 
 import pandas as pd
 
@@ -8,7 +8,7 @@ from setup import db_engine, session
 screen_status = {0: "off", 1: "on", 2: "locked", 3: "unlocked"}
 
 
-def get_screen_data(usernames: List) -> pd.DataFrame:
+def get_screen_data(usernames: Collection) -> pd.DataFrame:
     """
     Read the data from the screen table and return it in a dataframe.
 

test/test_participants.py

@@ -3,8 +3,8 @@ import unittest
 from participants.query_db import get_usernames
 
 
-class CallsFeatures(unittest.TestCase):
+class ParticipantsQuery(unittest.TestCase):
     def test_get_usernames(self):
         usernames_from_db = get_usernames()
-        print(usernames_from_db)
+        print(type(usernames_from_db))
         self.assertIsNotNone(usernames_from_db)