28 lines
1.0 KiB
Python
28 lines
1.0 KiB
Python
import unittest
|
|
|
|
import yaml
|
|
from pyprojroot import here
|
|
|
|
from machine_learning.features_sensor import *
|
|
|
|
|
|
class SensorFeaturesTest(unittest.TestCase):
|
|
@classmethod
|
|
def setUpClass(cls) -> None:
|
|
with open(here("machine_learning/config/minimal_features.yaml"), "r") as file:
|
|
cls.sensor_features_params = yaml.safe_load(file)
|
|
|
|
def test_yaml(self):
|
|
with open(here("machine_learning/config/minimal_features.yaml"), "r") as file:
|
|
sensor_features_params = yaml.safe_load(file)
|
|
self.assertIsInstance(sensor_features_params, dict)
|
|
self.assertIsInstance(sensor_features_params.get("grouping_variable"), str)
|
|
self.assertIsInstance(sensor_features_params.get("features"), dict)
|
|
self.assertIsInstance(
|
|
sensor_features_params.get("participants_usernames"), list
|
|
)
|
|
|
|
def test_participants_label(self):
|
|
sensor_features = SensorFeatures(**self.sensor_features_params)
|
|
self.assertRaises(ValueError, sensor_features.calculate_features)
|