From 325bb9e315ef0c8a63c8584ddc53d15accd71472 Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Tue, 3 Sep 2019 15:45:43 +0100
Subject: [PATCH] fix expected values in test_datatypes

---
 tests/test_datatypes.py | 30 +++++++++++++++---------------
 1 file changed, 15 insertions(+), 15 deletions(-)

diff --git a/tests/test_datatypes.py b/tests/test_datatypes.py
index 06b2b3597..98d5d387d 100644
--- a/tests/test_datatypes.py
+++ b/tests/test_datatypes.py
@@ -4,32 +4,32 @@ import finn.core.tensor as ten
 
 
 def test_datatypes():
-    assert ten.DataType.BIPOLAR.allowed(-1) is True
+    assert ten.DataType.BIPOLAR.allowed(-1)
     assert ten.DataType.BIPOLAR.allowed(0) is False
     assert ten.DataType.BINARY.allowed(-1) is False
-    assert ten.DataType.BINARY.allowed(-1) is True
-    assert ten.DataType.UINT2.allowed(2) is True
+    assert ten.DataType.BINARY.allowed(1)
+    assert ten.DataType.UINT2.allowed(2)
     assert ten.DataType.UINT2.allowed(10) is False
-    assert ten.DataType.UINT3.allowed(5) is True
+    assert ten.DataType.UINT3.allowed(5)
     assert ten.DataType.UINT3.allowed(-7) is False
-    assert ten.DataType.UINT4.allowed(15) is True
+    assert ten.DataType.UINT4.allowed(15)
     assert ten.DataType.UINT4.allowed(150) is False
-    assert ten.DataType.UINT8.allowed(150) is True
+    assert ten.DataType.UINT8.allowed(150)
     assert ten.DataType.UINT8.allowed(777) is False
-    assert ten.DataType.UINT16.allowed(14500) is True
+    assert ten.DataType.UINT16.allowed(14500)
     assert ten.DataType.UINT16.allowed(-1) is False
-    assert ten.DataType.UINT32.allowed(2 ** 10) is True
+    assert ten.DataType.UINT32.allowed(2 ** 10)
     assert ten.DataType.UINT32.allowed(-1) is False
-    assert ten.DataType.INT2.allowed(2) is True
+    assert ten.DataType.INT2.allowed(-1)
     assert ten.DataType.INT2.allowed(-10) is False
     assert ten.DataType.INT3.allowed(5) is False
-    assert ten.DataType.INT3.allowed(-2) is True
+    assert ten.DataType.INT3.allowed(-2)
     assert ten.DataType.INT4.allowed(15) is False
-    assert ten.DataType.INT4.allowed(-5) is True
+    assert ten.DataType.INT4.allowed(-5)
     assert ten.DataType.INT8.allowed(150) is False
-    assert ten.DataType.INT8.allowed(-127) is True
+    assert ten.DataType.INT8.allowed(-127)
     assert ten.DataType.INT16.allowed(-1.04) is False
-    assert ten.DataType.INT16.allowed(-7777) is True
+    assert ten.DataType.INT16.allowed(-7777)
     assert ten.DataType.INT32.allowed(7.77) is False
-    assert ten.DataType.INT32.allowed(-5) is True
-    assert ten.DataType.INT32.allowed(5) is True
+    assert ten.DataType.INT32.allowed(-5)
+    assert ten.DataType.INT32.allowed(5)
-- 
GitLab