🐛 Airbyte CDK: Fix flake errors in file-based CDK (#31771)

airbytehq · Oct 24, 2023 · c719137 · c719137
1 parent 2d552a9
commit c719137
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/...sources/file_based/availability_strategy/test_default_file_based_availability_strategy.py b/...sources/file_based/availability_strategy/test_default_file_based_availability_strategy.py
@@ -11,7 +11,7 @@
 )
 from airbyte_cdk.sources.file_based.config.file_based_stream_config import FileBasedStreamConfig
 from airbyte_cdk.sources.file_based.config.jsonl_format import JsonlFormat
-from airbyte_cdk.sources.file_based.exceptions import CheckAvailabilityError, CustomFileBasedException
+from airbyte_cdk.sources.file_based.exceptions import CustomFileBasedException
 from airbyte_cdk.sources.file_based.file_based_stream_reader import AbstractFileBasedStreamReader
 from airbyte_cdk.sources.file_based.file_types.file_type_parser import FileTypeParser
 from airbyte_cdk.sources.file_based.remote_file import RemoteFile
@@ -86,5 +86,5 @@ def test_catching_and_raising_custom_file_based_exception(self) -> None:
 
         # Invoke the check_availability_and_parsability method and check if it correctly handles the exception
         is_available, error_message = self._strategy.check_availability_and_parsability(self._stream, Mock(), Mock())
-        assert is_available == False
+        assert not is_available
         assert "Custom exception for testing." in error_message