Add custom coverage test

tthijm · Jun 12, 2024 · 2afde74 · 2afde74
1 parent 824a293
commit 2afde74
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/discord/utils.py b/discord/utils.py
@@ -71,6 +71,7 @@
 import typing
 import warnings
 import logging
+from cov import test, mark
 
 import yarl
 
@@ -1084,14 +1085,18 @@ def as_chunks(iterator: _Iter[T], max_size: int) -> _Iter[List[T]]:
 PY_312 = sys.version_info >= (3, 12)
 
 
+@test(3)
 def flatten_literal_params(parameters: Iterable[Any]) -> Tuple[Any, ...]:
     params = []
     literal_cls = type(Literal[0])
     for p in parameters:
         if isinstance(p, literal_cls):
+            mark(1)
             params.extend(p.__args__)
         else:
+            mark(2)
             params.append(p)
+    mark(0)
     return tuple(params)