test_parser.py

from __future__ import unicode_literals

import collections
import sys
if sys.version_info < (2, 7):
    from StringIO import StringIO
else:
    from io import StringIO
    
import pytest

import pytoml as toml


def test_name_of_fileobj_is_used_in_errors():
    source = StringIO("[")
    source.name = "<source>"
    error = pytest.raises(toml.TomlError, lambda: toml.load(source))
    assert error.value.filename == "<source>"


def test_when_fileobj_has_no_name_attr_then_repr_of_fileobj_is_used_in_errors():
    source = StringIO("[")
    error = pytest.raises(toml.TomlError, lambda: toml.load(source))
    assert error.value.filename == repr(source)


def test_object_pairs_hook():
    source = StringIO(u"""\
    [x.a]
    [x.b]
    [x.c]
    """)

    d = toml.load(source, object_pairs_hook=collections.defaultdict)
    assert isinstance(d, collections.defaultdict)
    assert isinstance(d['x'], collections.defaultdict)