Fix boolean comparison

Author: sdispater

tests/test_items.py

@@ -14,6 +14,7 @@
 from tomlkit import parse
 from tomlkit._compat import PY2
 from tomlkit.exceptions import NonExistentKey
+from tomlkit.items import Bool
 from tomlkit.items import InlineTable
 from tomlkit.items import Integer
 from tomlkit.items import Key
@@ -450,3 +451,22 @@ def test_deleting_inline_table_elemeent_does_not_leave_trailing_separator():
     table["baz"] = "boom"
 
     assert '{baz = "boom"}' == table.as_string()
+
+
+def test_booleans_comparison():
+    boolean = Bool(True, Trivia())
+
+    assert boolean
+
+    boolean = Bool(False, Trivia())
+
+    assert not boolean
+
+    s = """[foo]
+value = false
+"""
+
+    content = parse(s)
+
+    assert {"foo": {"value": False}} == content
+    assert {"value": False} == content["foo"]

tomlkit/items.py

@@ -496,7 +496,7 @@ class Bool(Item):
     A boolean literal.
     """
 
-    def __init__(self, t, trivia):  # type: (float, Trivia) -> None
+    def __init__(self, t, trivia):  # type: (int, Trivia) -> None
         super(Bool, self).__init__(trivia)
 
         self._value = bool(t)
@@ -515,6 +515,17 @@ def as_string(self):  # type: () -> str
     def _getstate(self, protocol=3):
         return self._value, self._trivia
 
+    def __bool__(self):
+        return self._value
+
+    __nonzero__ = __bool__
+
+    def __eq__(self, other):
+        if not isinstance(other, bool):
+            return NotImplemented
+
+        return other == self._value
+
 
 class DateTime(Item, datetime):
     """