9f81753c72087db0da0e5adc0191c0181cbecf3e,featuretools/tests/computational_backend/test_calculate_feature_matrix.py,,test_calls_progress_callback,#Any#,1218

Before Change

    // test with multiple jobs
    mock_progress_callback = MockProgressCallback()
    ft.calculate_feature_matrix(features, entityset=es, progress_callback=mock_progress_callback, n_jobs=3)

    assert np.isclose(mock_progress_callback.total_update, 100.0)
    assert np.isclose(mock_progress_callback.total_progress_percent, 100.0)

After Change

    mock_progress_callback = MockProgressCallback()

    with cluster() as (scheduler, [a, b]):
        dkwargs = {"cluster": scheduler["address"]}
        ft.calculate_feature_matrix(features,
                                    entityset=es,
                                    progress_callback=mock_progress_callback,
                                    dask_kwargs=dkwargs)

    assert np.isclose(mock_progress_callback.total_update, 100.0)
    assert np.isclose(mock_progress_callback.total_progress_percent, 100.0)

Instances