python-benedict/tests/core/test_dump.py

# -*- coding: utf-8 -*-

from benedict.core import dump as _dump
from decimal import Decimal
from six import PY3

import datetime as dt
import unittest


class dump_test_case(unittest.TestCase):

    @staticmethod
    def _rstrip_lines(s):
        return '\n'.join([line.rstrip() for line in s.splitlines()])

    def test_dump(self):
        d = {
            'a': {
                'b': {
                    'c': 1
                }
            }
        }
        r = """{
    "a": {
        "b": {
            "c": 1
        }
    }
}"""
        o = _dump(d)
        self.assertEqual(self._rstrip_lines(o), r)

    def test_dump_with_datetime(self):
        d = {
            'datetime': dt.datetime(2019, 6, 11),
        }
        r = """{
    "datetime": "2019-06-11T00:00:00"
}"""
        o = _dump(d)
        self.assertEqual(self._rstrip_lines(o), r)

    def test_dump_with_set(self):
        d = {
            'set': set([0, 1, 2, 3, 4, 5]),
        }
        r = """{
    "set": [
        0,
        1,
        2,
        3,
        4,
        5
    ]
}"""
        o = _dump(d)
        self.assertEqual(self._rstrip_lines(o), r)

    def test_dump_with_unsortable_keys(self):
        d = {
            None: None,
            0: 0,
            1: 1,
        }
        # must not raise TypeError
        _dump(d)
        d['dt'] = dt
        with self.assertRaises(TypeError):
            o = _dump(d, sort_keys=False, default=None)