Add custom coverage test

discord/utils.py

@@ -71,6 +71,7 @@
 import typing
 import warnings
 import logging
+from cov import test, mark
 
 import yarl
 
@@ -480,25 +481,28 @@ def find(predicate: Callable[[T], Any], iterable: _Iter[T], /) -> Union[Optional
         else _find(predicate, iterable)  # type: ignore
     )
 
-
+@test(3)
 def _get(iterable: Iterable[T], /, **attrs: Any) -> Optional[T]:
     # global -> local
     _all = all
     attrget = attrgetter
 
     # Special case the single element call
     if len(attrs) == 1:
+        mark(0)
         k, v = attrs.popitem()
         pred = attrget(k.replace('__', '.'))
         return next((elem for elem in iterable if pred(elem) == v), None)
 
     converted = [(attrget(attr.replace('__', '.')), value) for attr, value in attrs.items()]
     for elem in iterable:
         if _all(pred(elem) == value for pred, value in converted):
+            mark(1)
             return elem
+    mark(2)
     return None
 
-
+@test(4)
 async def _aget(iterable: AsyncIterable[T], /, **attrs: Any) -> Optional[T]:
     # global -> local
     _all = all
@@ -510,14 +514,18 @@ async def _aget(iterable: AsyncIterable[T], /, **attrs: Any) -> Optional[T]:
         pred = attrget(k.replace('__', '.'))
         async for elem in iterable:
             if pred(elem) == v:
+                mark(0)
                 return elem
+        mark(1)
         return None
 
     converted = [(attrget(attr.replace('__', '.')), value) for attr, value in attrs.items()]
 
     async for elem in iterable:
         if _all(pred(elem) == value for pred, value in converted):
+            mark(2)
             return elem
+    mark(3)
     return None