stress_at_work_analysis/test/test_proximity.py

34 lines
1.2 KiB
Python
Raw Permalink Normal View History

2021-08-17 10:51:51 +02:00
import unittest
2021-10-29 12:07:12 +02:00
from pyprojroot import here
2021-08-17 10:51:51 +02:00
from features.proximity import *
class ProximityFeatures(unittest.TestCase):
df_proximity = pd.DataFrame()
df_proximity_recoded = pd.DataFrame()
df_proximity_features = pd.DataFrame()
@classmethod
def setUpClass(cls) -> None:
2021-10-29 12:07:12 +02:00
cls.df_proximity = pd.read_csv(here("data/example_proximity.csv"))
2021-08-17 10:51:51 +02:00
cls.df_proximity["participant_id"] = 99
def test_recode_proximity(self):
self.df_proximity_recoded = recode_proximity(self.df_proximity)
self.assertIn("bool_prox_near", self.df_proximity_recoded)
# Is the recoded column present?
self.assertIn(True, self.df_proximity_recoded.bool_prox_near)
# Are there "near" values in the data?
self.assertIn(False, self.df_proximity_recoded.bool_prox_near)
# Are there "far" values in the data?
def test_count_proximity(self):
self.df_proximity_recoded = recode_proximity(self.df_proximity)
self.df_proximity_features = count_proximity(self.df_proximity_recoded)
print(self.df_proximity_features.columns)
self.assertCountEqual(
self.df_proximity_features.columns.to_list(), FEATURES_PROXIMITY
)