diff --git a/tests/test_sqlalchemy_athena.py b/tests/test_sqlalchemy_athena.py index 34e6483c..c3d2ce2b 100644 --- a/tests/test_sqlalchemy_athena.py +++ b/tests/test_sqlalchemy_athena.py @@ -89,7 +89,9 @@ def test_reflect_table(self, engine, conn): self.assertEqual(one_row.comment, "table comment") self.assertIn("location", one_row.dialect_options["awsathena"]) self.assertIn("compression", one_row.dialect_options["awsathena"]) - self.assertIsNotNone("location", one_row.dialect_options["awsathena"]["location"]) + self.assertIsNotNone( + "location", one_row.dialect_options["awsathena"]["location"] + ) @with_engine() def test_reflect_table_with_schema(self, engine, conn):