Improved pyupgrade script

docs/source/conf.py

@@ -154,7 +154,7 @@
     # Latex figure (float) alignment
     #
     # 'figure_align': 'htbp',
-    "preamble": "\setcounter{tocdepth}{4}"
+    "preamble": r"\setcounter{tocdepth}{4}"
 }
 
 # Grouping the document tree into LaTeX files. List of tuples

docs/source/run_autodoc.py

@@ -32,7 +32,7 @@ def replace_in_file(infile, replacements, outfile=None):
     if outfile is None:
         outfile = infile
 
-    with open(infile, "r") as fp:
+    with open(infile) as fp:
         content = fp.read()
 
     for key, value in replacements.items():

docs/sphinx_ext/simplify_typehints.py

@@ -48,8 +48,8 @@
 # replacement rules based on regular expressions
 REPLACEMENTS_REGEX = {
     # remove full package path and only leave the module/class identifier
-    "pde\.(\w+\.)*": "",
-    "typing\.": "",
+    r"pde\.(\w+\.)*": "",
+    r"typing\.": "",
 }
 
 

docs/sphinx_ext/toctree_filter.py

@@ -6,7 +6,7 @@
 class TocTreeFilter(TocTree):
     """directive to filter table-of-contents entries"""
 
-    hasPat = re.compile("^\s*:(.+):(.+)$")
+    hasPat = re.compile(r"^\s*:(.+):(.+)$")
 
     # Remove any entries in the content that we dont want and strip
     # out any filter prefixes that we want but obviously don't want the

scripts/format_code.sh

@@ -3,7 +3,7 @@
 
 echo "Upgrading python syntax..."
 pushd .. > /dev/null
-pyupgrade --py38-plus **/*.py
+find . -name '*.py' -exec pyupgrade --py38-plus {} +
 popd > /dev/null
 
 echo "Formating import statements..."

tests/tools/test_numba.py

@@ -40,7 +40,7 @@ def test_counter():
     assert int(c1) is 1
 
     c1 += 2
-    assert int(c1) is 3
+    assert int(c1) == 3
 
     c2 = Counter(3)
     assert c1 is not c2