test_sorts with dataclasses and namedtuples

Author: cclauss

tests/test_sorts.py

@@ -13,6 +13,8 @@
 0/1/2, ``bitonic_sort`` needs a power-of-two length, ``topological_sort`` works
 on a graph, and ``stalin_sort``/``wiggle_sort`` deliberately do not fully sort).
 """
+from dataclasses import dataclass
+typing import NamedTuple
 
 import pytest
 
@@ -69,6 +71,17 @@ def test_heap_sort():
     strand_sort,
 )
 
+@dataclass
+class Person:
+    name: str = "Bob"
+    age: int = 37
+    cost: float = 0.0
+
+class Dog(NamedTuple):
+    name: str = "Fido"
+    age: int = 5
+    weight: float = 15.5
+
 CASES = (
     [],
     [1],
@@ -79,6 +92,8 @@ def test_heap_sort():
     [5, 4, 3, 2, 1],
     [1, 2, 3, 4, 5],
     [-2, -2, 0, 0, 7, 7],
+    [Person(cost=100.0), Person(cost=-100.0), Person(name="Al")],
+    [Dog(weight=15.5), Dog(weight=15.1), Dog(name="Buddy")],
 )