コード例 #1
0
    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')
コード例 #2
0
    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