Skip to content

Commit

Permalink
Added doc strings to testing classes
Browse files Browse the repository at this point in the history
  • Loading branch information
Gabriel-Bottrill committed Nov 28, 2023
1 parent d3e8e74 commit d914ffd
Showing 1 changed file with 9 additions and 0 deletions.
9 changes: 9 additions & 0 deletions tests/devices/test_default_qutrit.py
Original file line number Diff line number Diff line change
Expand Up @@ -1404,6 +1404,9 @@ def test_density_matrix_all_wires(self, qutrit_device_2_wires, wires, expected):
@pytest.mark.jax
@pytest.mark.parametrize("use_jit", [True, False])
class TestQNodeIntegrationJax:
"""Integration tests for default.qutrit with JAX. This test ensures it integrates
properly with the PennyLane UI, in particular the QNode."""

def test_qutrit_circuit(self, tol, use_jit):
"""Test that the device provides the correct
result for a simple circuit."""
Expand Down Expand Up @@ -1597,6 +1600,9 @@ def circuit(a, b):

@pytest.mark.tf
class TestQNodeIntegrationTF:
"""Integration tests for default.qutrit with TensorFlow. This test ensures it integrates
properly with the PennyLane UI, in particular the QNode."""

def test_qutrit_circuit(self, tol):
"""Test that the device provides the correct
result for a simple circuit."""
Expand Down Expand Up @@ -1776,6 +1782,9 @@ def circuit(a, b):

@pytest.mark.torch
class TestQNodeIntegrationTorch:
"""Integration tests for default.qutrit with Torch. This test ensures it integrates
properly with the PennyLane UI, in particular the QNode."""

def test_qutrit_circuit(self, tol):
"""Test that the device provides the correct
result for a simple circuit."""
Expand Down

0 comments on commit d914ffd

Please sign in to comment.