diff --git a/nowcasting_datamodel/connection.py b/nowcasting_datamodel/connection.py index 9dc57089..a705e93a 100644 --- a/nowcasting_datamodel/connection.py +++ b/nowcasting_datamodel/connection.py @@ -49,7 +49,7 @@ def make_partitions(self): ): partition.__table__.create(bind=self.engine) - def drop_all(self): + def __drop_all(self): """Drop all partitions and tables""" # drop partitions for partition in self.partitions: diff --git a/tests/conftest.py b/tests/conftest.py index 8ac64929..43359d0d 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -52,7 +52,7 @@ def forecasts_all(db_session) -> List[ForecastSQL]: @pytest.fixture def db_connection(): - url = os.getenv("DB_URL", "sqlite:///test.db") + url = "sqlite:///test.db" connection = DatabaseConnection(url=url, echo=False) connection.create_all() @@ -80,7 +80,7 @@ def db_session(db_connection): @pytest.fixture def db_connection_pv(): - url = os.getenv("DB_URL_PV", "sqlite:///test_pv.db") + url = "sqlite:///test_pv.db" connection = DatabaseConnection(url=url, base=Base_PV) Base_PV.metadata.create_all(connection.engine)