Add __name__ to prettify test output

Author: awf

src/gfloat/block.py

@@ -47,8 +47,12 @@ def block_size_bytes(self) -> int:
         assert bits % 8 == 0
         return bits // 8
 
+    @property
+    def __name__(self) -> str:
+        return self.name
+
     def __str__(self) -> str:
-        return f"{self.name}"
+        return f"BlockFormatInfo:{self.name})"
 
 
 def decode_block(fi: BlockFormatInfo, block: Iterable[int]) -> Iterable[float]:

src/gfloat/formats.py

@@ -5,7 +5,7 @@
 
 #: FormatInfo for IEEE-754 Binary32 format
 format_info_binary32 = FormatInfo(
-    name="binary32",
+    name="format_info_binary32",
     k=32,
     precision=24,
     emax=127,
@@ -19,7 +19,7 @@
 
 #: FormatInfo for IEEE-754 Binary16 format
 format_info_binary16 = FormatInfo(
-    name="binary16",
+    name="format_info_binary16",
     k=16,
     precision=11,
     emax=15,
@@ -33,7 +33,7 @@
 
 #: FormatInfo for Google BFloat16 format
 format_info_bfloat16 = FormatInfo(
-    name="bfloat16",
+    name="format_info_bfloat16",
     k=16,
     precision=8,
     emax=127,
@@ -47,7 +47,7 @@
 
 #: FormatInfo for OCP E5M2 format
 format_info_ocp_e5m2 = FormatInfo(
-    name="ocp_e5m2",
+    name="format_info_ocp_e5m2",
     k=8,
     precision=3,
     emax=15,
@@ -61,7 +61,7 @@
 
 #: FormatInfo for OCP E4M3 format
 format_info_ocp_e4m3 = FormatInfo(
-    name="ocp_e4m3",
+    name="format_info_ocp_e4m3",
     k=8,
     precision=4,
     emax=8,
@@ -75,7 +75,7 @@
 
 #: FormatInfo for OCP MX E2M3 format
 format_info_ocp_e2m3 = FormatInfo(
-    name="ocp_e2m3",
+    name="format_info_ocp_e2m3",
     k=6,
     precision=4,
     emax=2,
@@ -89,7 +89,7 @@
 
 #: FormatInfo for OCP MX E3M2 format
 format_info_ocp_e3m2 = FormatInfo(
-    name="ocp_e3m2",
+    name="format_info_ocp_e3m2",
     k=6,
     precision=3,
     emax=4,
@@ -103,7 +103,7 @@
 
 #: FormatInfo for OCP MX E2M1 format
 format_info_ocp_e2m1 = FormatInfo(
-    name="ocp_e2m1",
+    name="format_info_ocp_e2m1",
     k=4,
     precision=2,
     emax=2,
@@ -117,7 +117,7 @@
 
 #: FormatInfo for OCP MX E8M0 format
 format_info_ocp_e8m0 = FormatInfo(
-    name="ocp_e8m0",
+    name="format_info_ocp_e8m0",
     k=8,
     precision=1,
     emax=127,
@@ -131,7 +131,7 @@
 
 #: FormatInfo for OCP MX INT8 format
 format_info_ocp_int8 = FormatInfo(
-    name="ocp_int8",
+    name="format_info_ocp_int8",
     k=8,
     precision=8,
     emax=0,
@@ -210,11 +210,11 @@ def format_info_p3109(precision: int) -> FormatInfo:
 # Block formats
 
 format_info_mxfp8_e5m2 = BlockFormatInfo(
-    "ocp_mxfp8_e5m2", format_info_ocp_e5m2, 32, format_info_ocp_e8m0
+    "format_info_mxfp8_e5m2", format_info_ocp_e5m2, 32, format_info_ocp_e8m0
 )
 
 format_info_mxfp8_e4m3 = BlockFormatInfo(
-    "ocp_mxfp8_e4m3", format_info_ocp_e4m3, 32, format_info_ocp_e8m0
+    "format_info_mxfp8_e4m3", format_info_ocp_e4m3, 32, format_info_ocp_e8m0
 )
 
 format_info_mxfp6_e3m2 = BlockFormatInfo(

src/gfloat/types.py

@@ -400,5 +400,9 @@ def is_all_subnormal(self) -> bool:
         """
         return (self.expBits == 0) and self.has_subnormals
 
+    @property
+    def __name__(self) -> str:
+        return self.name
+
     def __str__(self) -> str:
         return f"{self.name}"

test/test_block.py

@@ -7,7 +7,7 @@
 from gfloat.formats import *
 
 
-@pytest.mark.parametrize("fi", all_block_formats, ids=str)
+@pytest.mark.parametrize("fi", all_block_formats)
 def test_blocks(fi: BlockFormatInfo) -> None:
 
     vals = np.linspace(-37.0, 42.0, 32)

test/test_decode.py

@@ -183,15 +183,15 @@ def test_spot_check_ocp_int8() -> None:
     assert dec(0xFF) == -fi.smallest
 
 
-@pytest.mark.parametrize("fi", p3109_formats, ids=str)
+@pytest.mark.parametrize("fi", p3109_formats)
 def test_specials(fi: FormatInfo) -> None:
     assert fi.code_of_nan == 0x80
     assert fi.code_of_zero == 0x00
     assert fi.code_of_posinf == 0x7F
     assert fi.code_of_neginf == 0xFF
 
 
-@pytest.mark.parametrize("fi", all_formats, ids=str)
+@pytest.mark.parametrize("fi", all_formats)
 def test_specials_decode(fi: FormatInfo) -> None:
     dec = lambda v: decode_float(fi, v).fval
 
@@ -240,7 +240,7 @@ def test_except(v: int) -> None:
         decode_float(format_info_binary16, v)
 
 
-@pytest.mark.parametrize("fi", [fi for fi in all_formats if fi.bits <= 8], ids=str)
+@pytest.mark.parametrize("fi", [fi for fi in all_formats if fi.bits <= 8])
 def test_dense(fi: FormatInfo) -> None:
     fvs = [decode_float(fi, i) for i in range(0, 2**fi.bits)]
 

test/test_encode.py

@@ -8,7 +8,7 @@
 from gfloat.formats import *
 
 
-@pytest.mark.parametrize("fi", all_formats, ids=str)
+@pytest.mark.parametrize("fi", all_formats)
 def test_encode(fi: FormatInfo) -> None:
     dec = lambda v: decode_float(fi, v).fval
 
@@ -27,7 +27,7 @@ def test_encode(fi: FormatInfo) -> None:
         np.testing.assert_equal(fv2.fval, fv.fval)
 
 
-@pytest.mark.parametrize("fi", all_formats, ids=str)
+@pytest.mark.parametrize("fi", all_formats)
 def test_encode_edges(fi: FormatInfo) -> None:
     assert encode_float(fi, fi.max) == fi.code_of_max
 

test/test_microxcaling.py

@@ -25,7 +25,6 @@
         (ElemFormat.fp6_e3m2, format_info_ocp_e3m2),
         (ElemFormat.fp4_e2m1, format_info_ocp_e2m1),
     ],
-    ids=str,
 )
 def test_mx(
     mx_round: str,

test/test_round.py

@@ -117,7 +117,6 @@ def test_round_e4m3() -> None:
         format_info_ocp_e4m3,
         *p3109_formats,
     ],
-    ids=str,
 )
 def test_round(fi: FormatInfo) -> None:
     """
@@ -184,7 +183,7 @@ def test_round_ints(fi: FormatInfo, mldtype: Type) -> None:
         np.testing.assert_equal(val, mlval)
 
 
-@pytest.mark.parametrize("fi", all_formats, ids=str)
+@pytest.mark.parametrize("fi", all_formats)
 def test_round_roundtrip(fi: FormatInfo) -> None:
     if fi.bits <= 8:
         step = 1