RM-96 load with engine

bluelabsio · Jul 11, 2023 · 58b93f3 · 58b93f3
1 parent 0a37913
commit 58b93f3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/unit/db/postgres/test_loader.py b/tests/unit/db/postgres/test_loader.py
@@ -111,7 +111,7 @@ def test_load_from_fileobj_default_date_order_style(self,
         mock_Table.assert_called_with(mock_table,
                                       self.mock_meta,
                                       schema=mock_schema,
-                                      autoload_with=self.mock_db)
+                                      autoload_with=self.mock_db.engine)
         mock_conn = self.mock_db.engine.begin.return_value.__enter__.return_value
         mock_quote_value.assert_called_with(mock_conn, 'ISO, MDY')
         str_arg = str(mock_conn.execute.call_args.args[0])