Function test_write_options

python/pyarrow/tests/test_csv.py:414–444

()

def test_write_options():
    cls = WriteOptions
    opts = cls()

    check_options_class(
        cls, include_header=[True, False], delimiter=[',', '\t', '|'],
        quoting_style=['needed', 'none', 'all_valid'])

    assert opts.batch_size > 0
    opts.batch_size = 12345
    assert opts.batch_size == 12345

    opts = cls(batch_size=9876)
    assert opts.batch_size == 9876

    opts.validate()

    match = "WriteOptions: batch_size must be at least 1: 0"
    with pytest.raises(pa.ArrowInvalid, match=match):
        opts = cls()
        opts.batch_size = 0
        opts.validate()

    expected_repr_inner = """
    include_header=True,
    batch_size=0,
    delimiter=',',
    quoting_style='needed'"""

    assert repr(opts) == f"<pyarrow.csv.WriteOptions>({expected_repr_inner})"
    assert str(opts) == f"WriteOptions({expected_repr_inner})"


class BaseTestCSV(abc.ABC):