fix: ensure database cleanup after tests by dropping all tables

scholarX-Assesment · Jan 12, 2025 · 854ad8d · 854ad8d
1 parent f8d2b48
commit 854ad8d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tests/test_crud.py b/tests/test_crud.py
@@ -18,6 +18,7 @@ def db():
         yield db
     finally:
         db.close()
+        models.Base.metadata.drop_all(bind=engine) 
 
 def test_create_student(db):
     student = schemas.StudentCreate(name="Luke", email="luke@gmail.com", address="Kandy", contactNo="0771234567", cgpa=3.77)