diff --git a/tests/test_datatypes.py b/tests/test_datatypes.py new file mode 100644 index 0000000000000000000000000000000000000000..9ebd2f1eef195bfacd9f47ae2c924fb4123b38bf --- /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