Minor - Fix timestream test

Author: jaidisido

tests/conftest.py

@@ -1,6 +1,7 @@
 from datetime import datetime
 
 import boto3  # type: ignore
+import botocore.exceptions
 import pytest  # type: ignore
 
 import awswrangler as wr
@@ -303,6 +304,19 @@ def oracle_table():
     con.close()
 
 
+@pytest.fixture(scope="function")
+def timestream_database():
+    name = f"tbl_{get_time_str_with_random_suffix()}"
+    print(f"Timestream name: {name}")
+    wr.timestream.create_database(name)
+    yield name
+    try:
+        wr.timestream.delete_database(name)
+    except botocore.exceptions.ClientError as err:
+        if err.response["Error"]["Code"] == "ResourceNotFound":
+            pass
+
+
 @pytest.fixture(scope="function")
 def timestream_database_and_table():
     name = f"tbl_{get_time_str_with_random_suffix()}"

tests/test_timestream.py

@@ -246,26 +246,17 @@ def test_multimeasure_scenario(timestream_database_and_table):
     assert df.shape == (3, 6)
 
 
-def test_list_databases(timestream_database_and_table):
+def test_list_databases(timestream_database_and_table, timestream_database):
     dbs = wr.timestream.list_databases()
 
     assert timestream_database_and_table in dbs
-    dummy_db_name = f"{timestream_database_and_table}_2"
+    assert timestream_database in dbs
 
-    wr.timestream.create_database(dummy_db_name)
-    dbs_tmp = wr.timestream.list_databases()
-
-    assert timestream_database_and_table in dbs_tmp
-    assert dummy_db_name in dbs_tmp
-    assert len(dbs_tmp) == len(dbs) + 1
-
-    wr.timestream.delete_database(dummy_db_name)
+    wr.timestream.delete_database(timestream_database)
 
     dbs_tmp = wr.timestream.list_databases()
     assert timestream_database_and_table in dbs_tmp
-    assert dummy_db_name not in dbs_tmp
-    assert len(dbs_tmp) == len(dbs)
-    assert dbs_tmp == dbs
+    assert timestream_database not in dbs_tmp
 
 
 def test_list_tables(timestream_database_and_table):