File tree Expand file tree Collapse file tree 1 file changed +11
-1
lines changed
Expand file tree Collapse file tree 1 file changed +11
-1
lines changed Original file line number Diff line number Diff line change 11import os
2+ import sys
23import unittest
34
45from dune_client .file .base import CSVFile , NDJSONFile , JSONFile
@@ -151,8 +152,17 @@ def test_not_skip_empty_when_specified(self):
151152 # CSV empty files won't have any headers!
152153 self .file_manager ._write ([], writer , False )
153154 else :
154- with self .assertNoLogs ():
155+ if sys .version_info < (3 , 10 ):
156+ with self .assertRaises (FileNotFoundError ):
157+ self .file_manager ._load (writer )
158+ # assertNoLogs didn't exist till python 3.10, but we still support lower versions.
159+ # This is a bit of a hack, we write and then load to ensure the empty file was written.
155160 self .file_manager ._write ([], writer , False )
161+ # _load would return FileNotFoundError if it hadn't been written
162+ self .assertEqual (0 , len (self .file_manager ._load (writer )))
163+ else :
164+ with self .assertNoLogs ():
165+ self .file_manager ._write ([], writer , False )
156166
157167 self .file_manager ._load (writer )
158168
You can’t perform that action at this time.
0 commit comments