def test_write(self, mock_cropped_file_path, mock_np): subject_id = 'subjectA' data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9], [10, 11, 12]]) heart_rate_collection = HeartRateCollection(subject_id=subject_id, data=data) mock_cropped_file_path.return_value = file_path = 'path/to/file' HeartRateService.write(heart_rate_collection) mock_np.savetxt.assert_called_once_with(file_path, heart_rate_collection.data, fmt='%f')
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