From cb6c447848db466a31491141f81f37992d56584f Mon Sep 17 00:00:00 2001
From: Yaman Umuroglu <yamanu@xilinx.com>
Date: Wed, 21 Aug 2019 14:36:48 +0000
Subject: [PATCH] add some basic tests for datatypes

---
 tests/test_datatypes.py | 36 ++++++++++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)
 create mode 100644 tests/test_datatypes.py

diff --git a/tests/test_datatypes.py b/tests/test_datatypes.py
new file mode 100644
index 000000000..9ebd2f1ee
--- /dev/null
+++ b/tests/test_datatypes.py
@@ -0,0 +1,36 @@
+# -*- coding: utf-8 -*-
+
+import pytest
+
+import finn.core.tensor as ten
+
+def test_datatypes():
+  assert ten.DataType.BIPOLAR.allowed(-1) == True
+  assert ten.DataType.BIPOLAR.allowed(0) == False
+  assert ten.DataType.BINARY.allowed(-1) == False
+  assert ten.DataType.BINARY.allowed(-1) == True
+  assert ten.DataType.UINT2.allowed(2) == True
+  assert ten.DataType.UINT2.allowed(10) == False
+  assert ten.DataType.UINT3.allowed(5) == True
+  assert ten.DataType.UINT3.allowed(-7) == False
+  assert ten.DataType.UINT4.allowed(15) == True
+  assert ten.DataType.UINT4.allowed(150) == False
+  assert ten.DataType.UINT8.allowed(150) == True
+  assert ten.DataType.UINT8.allowed(777) == False
+  assert ten.DataType.UINT16.allowed(14500) == True
+  assert ten.DataType.UINT16.allowed(-1) == False
+  assert ten.DataType.UINT32.allowed(2 ** 10) == True
+  assert ten.DataType.UINT32.allowed(-1) == False
+  assert ten.DataType.INT2.allowed(2) == True
+  assert ten.DataType.INT2.allowed(-10) == False
+  assert ten.DataType.INT3.allowed(5) == False
+  assert ten.DataType.INT3.allowed(-2) == True
+  assert ten.DataType.INT4.allowed(15) == False
+  assert ten.DataType.INT4.allowed(-5) == True
+  assert ten.DataType.INT8.allowed(150) == False
+  assert ten.DataType.INT8.allowed(-127) == True
+  assert ten.DataType.INT16.allowed(-1.04) == False
+  assert ten.DataType.INT16.allowed(-7777) == True
+  assert ten.DataType.INT32.allowed(7.77) == False
+  assert ten.DataType.INT32.allowed(-5) == True
+  assert ten.DataType.INT32.allowed(5) == True
-- 
GitLab