testpython.git

@@ -12,10 +12,12 @@ from typing import Union, Optional, Tuple, List, Callable

 def profile(sort: Union[Tuple, List, str]='cumulative',

-            lines_to_print: Union[int, float]=20,

+            lines_to_print: float=20,

             strip_dirs: bool=True,

             filename: Optional[str]=None) -> Callable:

     """A decorator which profiles a callable.

+    If lines_to_print is an int, it's the number of lines. If it's a float

+    then it's the percentage of total lines. (Eg., .2 = 20%)

     Output goes to stdout if filename is None, otherwise to filename."""

     def outer(fun: Callable) -> Callable:

         def inner(*args, **kwargs):

@@ -43,5 +45,5 @@ def profile(sort: Union[Tuple, List, str]='cumulative',

     if hasattr(sort, '__call__'):

         fun = sort

         sort = 'cumulative'

-        outer = outer(fun)

+        outer = outer(fun)  # type: ignore

     return outer

@@ -38,7 +38,7 @@ def sitesize(url: str) -> Tuple[str, Union[int, str]]:

 def run() -> None:

     # Network bound: Use max(); CPU bound: Use min()

-    pool = ThreadPool(max(len(sites), os.cpu_count()))

+    pool = ThreadPool(max(len(sites), os.cpu_count() is None and 1 or os.cpu_count()))  # type: ignore

     for result in pool.imap_unordered(sitesize, sites):

         print(result)