def METHOD_NAME(
    pick_and_freeze_samples_small, sobol_object_input_samples_small
):
    """Test the `generate_pick_and_test_samples` function."""
    # Prepare
    A_samples, B_samples, C_samples = pick_and_freeze_samples_small
    A_test, B_test, C_test_generator, _ = sobol_object_input_samples_small
    # Act
    assert np.allclose(A_samples, np.around(A_test, 3))
    assert np.allclose(B_samples, np.around(B_test, 3))
    for i in range(3):
        C_test = next(C_test_generator)
        assert np.allclose(C_samples[i, :, :], np.around(C_test, 3))