Expect operation to succeed

Jesse Whitehouse · Jesse Whitehouse · commit fc06ef87e27d · 2022-11-23T14:09:47.000-06:00
Signed-off-by: Jesse Whitehouse &lt;jesse.whitehouse@databricks.com&gt;
diff --git a/tests/e2e/driver_tests.py b/tests/e2e/driver_tests.py
@@ -678,12 +678,9 @@ def test_staging_ingestion_life_cycle(self):
         
         remove_query = f"REMOVE 'stage://tmp/{self.staging_ingestion_user}/tmp/11/15/file1.csv'"
 
-
-        # Should raise an exception because REMOVE is not yet implemented
-        with pytest.raises(Error):
-            with self.connection() as conn:
-                cursor = conn.cursor()
-                cursor.execute(remove_query)
+        with self.connection() as conn:
+            cursor = conn.cursor()
+            cursor.execute(remove_query)
             
         os.remove(temp_path)
         os.remove(new_temp_path)
