More coverage.

Author: nwittler

test/test_quantity.py

@@ -69,6 +69,11 @@ def test_qty_str() -> None:
     assert str(gate_time) == "5.325 ns "
 
 
+@pytest.mark.unit
+def test_qty_item() -> None:
+    assert gate_time[0] == gate_time.numpy()
+
+
 @pytest.mark.unit
 def test_qty_set() -> None:
     gate_time.set_value(7e-9)
@@ -98,6 +103,16 @@ def test_qty_matrix_str() -> None:
     assert str(matrix) == "0.000  1.000  1.000  0.000  "
 
 
+@pytest.mark.unit
+def test_qty_matrix_list() -> None:
+    assert matrix.tolist() == [[0, 1], [1, 0]]
+
+
+@pytest.mark.unit
+def test_qty_matrix_item() -> None:
+    assert matrix[0][1] == 1
+
+
 @pytest.mark.unit
 def test_qty_matrix_set() -> None:
     matrix.set_value([[1.0, 0.0], [0.0, 1.0]])
@@ -165,3 +180,16 @@ def get_and_set() -> None:
         opt_val = a.get_opt_value()
         a.set_opt_value(opt_val)
         np.testing.assert_allclose(a, val)
+
+
+@pytest.mark.unit
+def test_qty_matrix_float() -> None:
+    with pytest.raises(NotImplementedError):
+        float(matrix)
+
+
+@pytest.mark.unit
+def test_pihalf() -> None:
+    "Don't use prefixes with pi."
+    pihalf = Quantity(1500, unit="pi")
+    assert str(pihalf) == "1500.0 pi "