diff --git a/target_postgres/sinks.py b/target_postgres/sinks.py index cce1f8b..5dad544 100644 --- a/target_postgres/sinks.py +++ b/target_postgres/sinks.py @@ -153,8 +153,6 @@ def _do_copy( for column in columns } - raise NotImplementedError("This method is not implemented yet.") - # Use copy to run the copy statement. # https://www.psycopg.org/psycopg3/docs/basic/copy.html with connection.connection.cursor().copy(copy_statement) as copy: # type: ignore[attr-defined] @@ -172,6 +170,8 @@ def _do_copy( copy.write_row(processed_row) + raise NotImplementedError("This method is not implemented yet.") + def bulk_insert_records( # type: ignore[override] self, table: sa.Table, diff --git a/target_postgres/tests/conftest.py b/target_postgres/tests/conftest.py new file mode 100644 index 0000000..55450d9 --- /dev/null +++ b/target_postgres/tests/conftest.py @@ -0,0 +1,8 @@ +"""Configuration for pytest.""" + +import os + + +def pytest_report_header(): + """Add environment variables to the pytest report header.""" + return [f"{var}: value" for var in os.environ if var.startswith("TARGET_POSTGRES")]