Function test_used_on_data_range

python/ray/data/tests/test_telemetry.py:29–37

(reset_usage_lib, callsite: TelemetryCallsite)

@pytest.mark.parametrize("callsite", list(TelemetryCallsite))
def test_used_on_data_range(reset_usage_lib, callsite: TelemetryCallsite):
    def _call_data_range():
        data.range(10)

    check_library_usage_telemetry(
        _call_data_range,
        callsite=callsite,
        expected_library_usages=[{"core", "dataset"}],
    )


if __name__ == "__main__":