attrs/tests/test_dark_magic.py

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

from __future__ import absolute_import, division, print_function


import attr

from attr._make import Attribute, NOTHING


@attr.s
class C1(object):
    x = attr.ib()
    y = attr.ib()


foo = None


@attr.s()
class C2(object):
    x = attr.ib(default_value=foo)
    y = attr.ib(default_factory=list)


class TestDarkMagic(object):
    """
    Integration tests.
    """
    def test_ls(self):
        """
        `attr.ls` works.
        """
        assert [
            Attribute(name="x", default_value=None, default_factory=NOTHING),
            Attribute(name="y", default_value=NOTHING, default_factory=list),
        ] == attr.ls(C2)

    def test_to_dict(self):
        """
        `attr.to_dict` works.
        """
        assert {
            "x": 1,
            "y": 2,
        } == attr.to_dict(C1(x=1, y=2))