def METHOD_NAME(self, rai_insights):
    assert rai_insights._large_test is None
    assert rai_insights._large_predict_output is None
    assert rai_insights._large_predict_proba_output is None
    dataset = rai_insights._get_dataset()
    assert not dataset.is_large_data_scenario
    assert not dataset.use_entire_test_data
    filtered_small_data = rai_insights.get_filtered_test_data(
        [], [], use_entire_test_data=False)
    assert len(filtered_small_data) == len(rai_insights.test)
    filtered_large_data = rai_insights.get_filtered_test_data(
        [], [], use_entire_test_data=True)
    assert len(filtered_large_data) == len(rai_insights.test)