4242 system_test_extras = extras ,
4343 system_test_extras_by_python = extras_by_python ,
4444)
45- s .move (templated_files , excludes = [
46- # sqlalchemy-bigquery was originally licensed MIT
47- "LICENSE" ,
48- "docs/multiprocessing.rst" ,
49- # exclude gh actions as credentials are needed for tests
50- ".github/workflows" ,
51- "README.rst" ,
52- ])
45+ s .move (
46+ templated_files ,
47+ excludes = [
48+ # sqlalchemy-bigquery was originally licensed MIT
49+ "LICENSE" ,
50+ "docs/multiprocessing.rst" ,
51+ # exclude gh actions as credentials are needed for tests
52+ ".github/workflows" ,
53+ "README.rst" ,
54+ ],
55+ )
5356
5457# ----------------------------------------------------------------------------
5558# Fixup files
5962 [".coveragerc" ],
6063 "google/cloud/__init__.py" ,
6164 "sqlalchemy_bigquery/requirements.py" ,
62- )
65+ )
6366
6467s .replace (
6568 ["noxfile.py" ],
7578
7679
7780s .replace (
78- ["noxfile.py" ], "--cov=google" , "--cov=sqlalchemy_bigquery" ,
81+ ["noxfile.py" ],
82+ "--cov=google" ,
83+ "--cov=sqlalchemy_bigquery" ,
7984)
8085
8186
8287s .replace (
83- ["noxfile.py" ],
88+ ["noxfile.py" ],
8489 "\+ SYSTEM_TEST_EXTRAS" ,
8590 "" ,
8691)
8792
8893
8994s .replace (
9095 ["noxfile.py" ],
91- ''' "protobuf",
92- # dependency of grpc''' ,
93- ''' "protobuf",
94- "sqlalchemy>=2.0.0 ,<2.1",
95- # dependency of grpc''' ,
96+ """ "protobuf",
97+ # dependency of grpc""" ,
98+ """ "protobuf",
99+ "sqlalchemy>=1.4 ,<2.1",
100+ # dependency of grpc""" ,
96101)
97102
98103
99104s .replace (
100105 ["noxfile.py" ],
101106 r"def default\(session\)" ,
102- "def default(session, install_extras=True)" ,
107+ "def default(session, install_extras=True)" ,
103108)
104109
105110
106-
107-
108111def place_before (path , text , * before_text , escape = None ):
109112 replacement = "\n " .join (before_text ) + "\n " + text
110113 if escape :
111114 for c in escape :
112- text = text .replace (c , ' \\ ' + c )
115+ text = text .replace (c , " \\ " + c )
113116 s .replace ([path ], text , replacement )
114117
118+
115119place_before (
116120 "noxfile.py" ,
117121 "SYSTEM_TEST_PYTHON_VERSIONS=" ,
118122 "" ,
119- "# We're using two Python versions to test with sqlalchemy>=2.0.0 " ,
123+ "# We're using two Python versions to test with sqlalchemy>=1.4 " ,
120124)
121125
122126place_before (
@@ -126,15 +130,15 @@ def place_before(path, text, *before_text, escape=None):
126130)
127131
128132
129- install_logic = '''
133+ install_logic = """
130134 if install_extras and session.python in ["3.11", "3.12"]:
131135 install_target = ".[geography,alembic,tests,bqstorage]"
132136 elif install_extras:
133137 install_target = ".[all]"
134138 else:
135139 install_target = "."
136140 session.install("-e", install_target, "-c", constraints_path)
137- '''
141+ """
138142
139143place_before (
140144 "noxfile.py" ,
@@ -163,7 +167,7 @@ def compliance(session):
163167 session.skip("Compliance tests were not found")
164168
165169 session.install("--pre", "grpcio")
166- session.install("--pre", "--no-deps", "--upgrade", "sqlalchemy>=2.0.0 ,<2.1")
170+ session.install("--pre", "--no-deps", "--upgrade", "sqlalchemy>=1.4 ,<2.1")
167171 session.install(
168172 "mock",
169173 "pytest",
@@ -205,12 +209,11 @@ def compliance(session):
205209'''
206210
207211place_before (
208- "noxfile.py" ,
209- "@nox.session(python=DEFAULT_PYTHON_VERSION)\n "
210- "def cover(session):" ,
211- compliance ,
212- escape = "()" ,
213- )
212+ "noxfile.py" ,
213+ "@nox.session(python=DEFAULT_PYTHON_VERSION)\n " "def cover(session):" ,
214+ compliance ,
215+ escape = "()" ,
216+ )
214217
215218s .replace (["noxfile.py" ], '"alabaster"' , '"alabaster", "geoalchemy2", "shapely"' )
216219
@@ -266,11 +269,10 @@ def system_noextras(session):
266269
267270place_before (
268271 "noxfile.py" ,
269- "@nox.session(python=SYSTEM_TEST_PYTHON_VERSIONS[-1])\n "
270- "def compliance(session):" ,
272+ "@nox.session(python=SYSTEM_TEST_PYTHON_VERSIONS[-1])\n " "def compliance(session):" ,
271273 system_noextras ,
272274 escape = "()[]" ,
273- )
275+ )
274276
275277
276278# Add DB config for SQLAlchemy dialect test suite.
@@ -287,7 +289,7 @@ def system_noextras(session):
287289[tool:pytest]
288290addopts= --tb native -v -r fxX -p no:warnings
289291python_files=tests/*test_*.py
290- """
292+ """ ,
291293)
292294
293295# ----------------------------------------------------------------------------
@@ -298,7 +300,7 @@ def system_noextras(session):
298300python .py_samples (skip_readmes = True )
299301
300302s .replace (
301- ["./samples/snippets/noxfile.py" ],
303+ ["./samples/snippets/noxfile.py" ],
302304 """session.install\("-e", _get_repo_root\(\)\)""" ,
303305 """session.install("-e", _get_repo_root())
304306 else:
0 commit comments