tests/test_data.py

import pytest

from datasets import load_dataset