def test_load_raw(self, mock_get_raw_file_path, mock_load,
mock_remove_repeats):
subject_id = 'subjectA'
mock_get_raw_file_path.return_value = returned_raw_path = 'path/to/file'
mock_load.return_value = data = np.array([[1, 2, 3], [4, 5, 6]])
mock_remove_repeats.return_value = data
expected_heart_rate_collection = HeartRateCollection(
subject_id=subject_id, data=data)
returned_heart_rate_collection = HeartRateService.load_raw(subject_id)
TestHelper.assert_models_equal(self, expected_heart_rate_collection,
returned_heart_rate_collection)
mock_get_raw_file_path.assert_called_once_with(subject_id)
mock_load.assert_called_once_with(returned_raw_path, ',')
mock_remove_repeats.assert_called_once_with(data)
def crop_all(subject_id):
# psg_raw_collection = PSGService.read_raw(subject_id) # Used to extract PSG details from the reports
psg_raw_collection = PSGService.read_precleaned(
subject_id) # Loads already extracted PSG data
motion_collection = MotionService.load_raw(subject_id)
heart_rate_collection = HeartRateService.load_raw(subject_id)
valid_interval = RawDataProcessor.get_intersecting_interval(
[psg_raw_collection, motion_collection, heart_rate_collection])
psg_raw_collection = PSGService.crop(psg_raw_collection,
valid_interval)
motion_collection = MotionService.crop(motion_collection,
valid_interval)
heart_rate_collection = HeartRateService.crop(heart_rate_collection,
valid_interval)
PSGService.write(psg_raw_collection)
MotionService.write(motion_collection)
HeartRateService.write(heart_rate_collection)
ActivityCountService.build_activity_counts_without_matlab(
subject_id, motion_collection.data
) # Builds activity counts with python, not MATLAB