fix test

snowflakedb · Aug 28, 2024 · 13ea322 · 13ea322
1 parent 3da5e48
commit 13ea322
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/tests/integ/modin/types/test_timedelta.py b/tests/integ/modin/types/test_timedelta.py
@@ -2,15 +2,11 @@
 # Copyright (c) 2012-2024 Snowflake Computing Inc. All rights reserved.
 #
 import datetime
-import logging
 
 import modin.pandas as pd
 import pandas as native_pd
 import pytest
 
-from snowflake.snowpark.modin.plugin._internal.snowpark_pandas_types import (
-    TIMEDELTA_WARNING_MESSAGE,
-)
 from tests.integ.modin.sql_counter import sql_count_checker
 from tests.integ.modin.utils import (
     assert_series_equal,
@@ -21,13 +17,11 @@
 
 
 @sql_count_checker(query_count=1)
-def test_create_timedelta_column_from_pandas_timedelta(caplog):
+def test_create_timedelta_column_from_pandas_timedelta():
     pandas_df = native_pd.DataFrame(
         {"timedelta_column": [native_pd.Timedelta(nanoseconds=1)], "int_column": [3]}
     )
-    with caplog.at_level(logging.DEBUG):
-        snow_df = pd.DataFrame(pandas_df)
-    assert TIMEDELTA_WARNING_MESSAGE in caplog.text
+    snow_df = pd.DataFrame(pandas_df)
     eval_snowpark_pandas_result(snow_df, pandas_df, lambda df: df)