import pytest
from privacy_tuner.input.text_loader import TextLoader


def test_text_loader_interface():
    loader = TextLoader()
    with pytest.raises(NotImplementedError):
        loader.load("dummy_path")

    # 测试示例方法
    sample = loader.load_sample()
    assert sample.shape == (2, 2)
    assert list(sample.columns) == ['text', 'label']