From 81b053b11babc8b609087132a77cdc8adbe74808 Mon Sep 17 00:00:00 2001 From: Krrish Dholakia Date: Sat, 30 Nov 2024 18:47:33 -0800 Subject: [PATCH] test: update test --- tests/proxy_unit_tests/test_key_generate_prisma.py | 11 +---------- 1 file changed, 1 insertion(+), 10 deletions(-) diff --git a/tests/proxy_unit_tests/test_key_generate_prisma.py b/tests/proxy_unit_tests/test_key_generate_prisma.py index 26d37a772..0832c1137 100644 --- a/tests/proxy_unit_tests/test_key_generate_prisma.py +++ b/tests/proxy_unit_tests/test_key_generate_prisma.py @@ -1345,17 +1345,8 @@ def test_generate_and_update_key(prisma_client): ) current_time = datetime.now(timezone.utc) - print( - "days between now and budget_reset_at", - (budget_reset_at - current_time).days, - ) # assert budget_reset_at is 30 days from now - assert ( - abs( - (budget_reset_at - current_time).total_seconds() - 30 * 24 * 60 * 60 - ) - <= 10 - ) + assert 31 >= (budget_reset_at - current_time).days >= 29 # cleanup - delete key delete_key_request = KeyRequest(keys=[generated_key])