c19000b012bba5fa7ae54fadc953f5b679dead4f,featurex/tests/test_extractors.py,,test_optical_flow_extractor,#,145

Before Change

    ext = DenseOpticalFlowExtractor()
    timeline = stim.extract([ext])
    df = timeline.to_df()
    assert df.shape == (168, 4)
    target = df.query("name=="total_flow" & onset==3.0")["value"].values
    assert np.isclose(target, 86248.05)

After Change

    video_dir = join(get_test_data_path(), "video")
    stim = VideoStim(join(video_dir, "small.mp4"))
    result = DenseOpticalFlowExtractor().extract(stim).to_df()
    target = result.query("onset==3.0")["total_flow"]
    assert np.isclose(target, 86248.05)


@pytest.mark.skipif(""INDICO_APP_KEY" not in os.environ")

Instances