diff --git a/tests/brevitas/test_quant_tensor.py b/tests/brevitas/test_quant_tensor.py new file mode 100644 index 000000000..2054d8a28 --- /dev/null +++ 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)