Add schema search path setting in initialize_schema function in tests

Author: manuelwedler

tests/helpers.py

@@ -243,9 +243,11 @@ def dummy_verified_contract():
     return VerifiedContract.dummy()
 
 
-def initialize_schema(connection):
+def initialize_schema(connection, schema_name="public"):
     with open('./database.sql', 'r') as schema_file:
         schema = schema_file.read()
 
     with connection.cursor() as cursor:
+        # Set the schema search path
+        cursor.execute(f"SET search_path = {schema_name}")
         cursor.execute(schema)