fix: add retry for test_clean_up_via_context_manager (#1627)

sycai · web-flow · commit f580a569994e · 2025-04-21T17:52:17.000-07:00
diff --git a/packages/bigframes/tests/system/large/test_session.py b/packages/bigframes/tests/system/large/test_session.py
@@ -151,6 +151,7 @@ def test_clean_up_by_session_id():
         pytest.param(bigframes.connect, id="connect-method"),
     ],
 )
+@pytest.mark.flaky(retries=3)
 def test_clean_up_via_context_manager(session_creator):
     # we will create two tables and confirm that they are deleted
     # when the session is closed

Original file line number	Diff line number	Diff line change
`@@ -151,6 +151,7 @@ def test_clean_up_by_session_id():`
`151`	`151`	`pytest.param(bigframes.connect, id="connect-method"),`
`152`	`152`	`],`
`153`	`153`	`)`
	`154`	`+@pytest.mark.flaky(retries=3)`
`154`	`155`	`def test_clean_up_via_context_manager(session_creator):`
`155`	`156`	`# we will create two tables and confirm that they are deleted`
`156`	`157`	`# when the session is closed`