Added test

Xilinx · Mar 22, 2024 · 1f61f28 · 1f61f28
1 parent 5c9cb19
commit 1f61f28
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/tests/brevitas/test_quant_tensor.py b/tests/brevitas/test_quant_tensor.py
@@ -0,0 +1,12 @@
+import torch
+
+from brevitas.quant_tensor import QuantTensor
+from brevitas.quant_tensor import QuantTensorBase
+
+
+def test_qt():
+    v = torch.randn(1)
+    qt = QuantTensor(v, v, v, v, True, True)
+    assert isinstance(qt, QuantTensor)
+    assert isinstance(qt, tuple)
+    assert isinstance(qt, QuantTensorBase)