Skip to content

Commit 6434584

Browse files
committed
add import test
1 parent 0abf98c commit 6434584

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

qdp/qdp-python/tests/test_bindings.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,12 @@ def test_import():
3535
assert hasattr(mahout_qdp, "QdpEngine")
3636
assert hasattr(mahout_qdp, "QuantumTensor")
3737

38+
# Test that QdpEngine has the new encode_from_tensorflow method
39+
from mahout_qdp import QdpEngine
40+
41+
assert hasattr(QdpEngine, "encode_from_tensorflow")
42+
assert callable(getattr(QdpEngine, "encode_from_tensorflow"))
43+
3844

3945
@pytest.mark.gpu
4046
def test_encode():

0 commit comments

Comments
 (0)