Enable 22

pandas/tests/extension/test_sparse.py

@@ -70,7 +70,7 @@ def gen(count):
  for _ in range(count):
  yield SparseArray(make_data(request.param), fill_value=request.param)
 
- yield gen
+ return gen
 
 
 @pytest.fixture(params=[0, np.nan])

pandas/tests/frame/test_query_eval.py

@@ -1187,7 +1187,7 @@ def df(self):
  by backticks. The last two columns cannot be escaped by backticks
  and should raise a ValueError.
  """
- yield DataFrame(
+ return DataFrame(
  {
  "A": [1, 2, 3],
  "B B": [3, 2, 1],

pandas/tests/io/formats/test_to_latex.py

@@ -771,7 +771,7 @@ def df_with_symbols(self):
  """Dataframe with special characters for testing chars escaping."""
  a = "a"
  b = "b"
- yield DataFrame({"co$e^x$": {a: "a", b: "b"}, "co^l1": {a: "a", b: "b"}})
+ return DataFrame({"co$e^x$": {a: "a", b: "b"}, "co^l1": {a: "a", b: "b"}})
 
  def test_to_latex_escape_false(self, df_with_symbols):
  result = df_with_symbols.to_latex(escape=False)
@@ -1010,7 +1010,7 @@ class TestToLatexMultiindex:
  @pytest.fixture
  def multiindex_frame(self):
  """Multiindex dataframe for testing multirow LaTeX macros."""
- yield DataFrame.from_dict(
+ return DataFrame.from_dict(
  {
  ("c1", 0): Series({x: x for x in range(4)}),
  ("c1", 1): Series({x: x + 4 for x in range(4)}),
@@ -1023,7 +1023,7 @@ def multiindex_frame(self):
  @pytest.fixture
  def multicolumn_frame(self):
  """Multicolumn dataframe for testing multicolumn LaTeX macros."""
- yield DataFrame(
+ return DataFrame(
  {
  ("c1", 0): {x: x for x in range(5)},
  ("c1", 1): {x: x + 5 for x in range(5)},

pandas/tests/io/pytables/test_compat.py

@@ -36,7 +36,7 @@ def pytables_hdf5_file(tmp_path):
  t.row[key] = value
  t.row.append()
 
- yield path, objname, pd.DataFrame(testsamples)
+ return path, objname, pd.DataFrame(testsamples)
 
 
 class TestReadPyTablesHDF5:

pandas/tests/io/test_sql.py

@@ -620,13 +620,13 @@ def mysql_pymysql_engine():
 def mysql_pymysql_engine_iris(mysql_pymysql_engine, iris_path):
  create_and_load_iris(mysql_pymysql_engine, iris_path)
  create_and_load_iris_view(mysql_pymysql_engine)
- yield mysql_pymysql_engine
+ return mysql_pymysql_engine
 
 
 @pytest.fixture
 def mysql_pymysql_engine_types(mysql_pymysql_engine, types_data):
  create_and_load_types(mysql_pymysql_engine, types_data, "mysql")
- yield mysql_pymysql_engine
+ return mysql_pymysql_engine
 
 
 @pytest.fixture
@@ -667,13 +667,13 @@ def postgresql_psycopg2_engine():
 def postgresql_psycopg2_engine_iris(postgresql_psycopg2_engine, iris_path):
  create_and_load_iris(postgresql_psycopg2_engine, iris_path)
  create_and_load_iris_view(postgresql_psycopg2_engine)
- yield postgresql_psycopg2_engine
+ return postgresql_psycopg2_engine
 
 
 @pytest.fixture
 def postgresql_psycopg2_engine_types(postgresql_psycopg2_engine, types_data):
  create_and_load_types(postgresql_psycopg2_engine, types_data, "postgres")
- yield postgresql_psycopg2_engine
+ return postgresql_psycopg2_engine
 
 
 @pytest.fixture
@@ -713,7 +713,7 @@ def postgresql_adbc_iris(postgresql_adbc_conn, iris_path):
  except mgr.ProgrammingError: # note arrow-adbc issue 1022
  conn.rollback()
  create_and_load_iris_view(conn)
- yield conn
+ return conn
 
 
 @pytest.fixture
@@ -730,7 +730,7 @@ def postgresql_adbc_types(postgresql_adbc_conn, types_data):
 
  create_and_load_types_postgresql(conn, new_data)
 
- yield conn
+ return conn
 
 
 @pytest.fixture
@@ -784,7 +784,7 @@ def sqlite_str_iris(sqlite_str, iris_path):
 def sqlite_engine_iris(sqlite_engine, iris_path):
  create_and_load_iris(sqlite_engine, iris_path)
  create_and_load_iris_view(sqlite_engine)
- yield sqlite_engine
+ return sqlite_engine
 
 
 @pytest.fixture
@@ -805,7 +805,7 @@ def sqlite_str_types(sqlite_str, types_data):
 @pytest.fixture
 def sqlite_engine_types(sqlite_engine, types_data):
  create_and_load_types(sqlite_engine, types_data, "sqlite")
- yield sqlite_engine
+ return sqlite_engine
 
 
 @pytest.fixture
@@ -845,7 +845,7 @@ def sqlite_adbc_iris(sqlite_adbc_conn, iris_path):
  except mgr.ProgrammingError:
  conn.rollback()
  create_and_load_iris_view(conn)
- yield conn
+ return conn
 
 
 @pytest.fixture
@@ -867,7 +867,7 @@ def sqlite_adbc_types(sqlite_adbc_conn, types_data):
  create_and_load_types_sqlite3(conn, new_data)
  conn.commit()
 
- yield conn
+ return conn
 
 
 @pytest.fixture
@@ -881,14 +881,14 @@ def sqlite_buildin():
 def sqlite_buildin_iris(sqlite_buildin, iris_path):
  create_and_load_iris_sqlite3(sqlite_buildin, iris_path)
  create_and_load_iris_view(sqlite_buildin)
- yield sqlite_buildin
+ return sqlite_buildin
 
 
 @pytest.fixture
 def sqlite_buildin_types(sqlite_buildin, types_data):
  types_data = [tuple(entry.values()) for entry in types_data]
  create_and_load_types_sqlite3(sqlite_buildin, types_data)
- yield sqlite_buildin
+ return sqlite_buildin
 
 
 mysql_connectable = [

pyproject.toml

@@ -246,6 +246,7 @@ select = [
  "PT016",
  "PT020",
  "PT021",
+ "PT022",
  # misc lints
  "PIE",
  # flake8-pyi