@@ -119,6 +119,7 @@ func TestRandomDatums(t *testing.T) {
119119 fileName := "TestRandomDatums.parquet"
120120 f , err := os .CreateTemp ("" , fileName )
121121 require .NoError (t , err )
122+ defer removeFileUnlessFailed (t , f )
122123
123124 schemaDef , err := NewSchema (sch .columnNames , sch .columnTypes )
124125 require .NoError (t , err )
@@ -520,6 +521,7 @@ func TestBasicDatums(t *testing.T) {
520521 fileName := "TestBasicDatums.parquet"
521522 f , err := os .CreateTemp ("" , fileName )
522523 require .NoError (t , err )
524+ defer removeFileUnlessFailed (t , f )
523525
524526 schemaDef , err := NewSchema (tc .sch .columnNames , tc .sch .columnTypes )
525527 require .NoError (t , err )
@@ -627,6 +629,7 @@ func optionsTest(t *testing.T, opt Option, testFn func(t *testing.T, reader *fil
627629 fileName := "OptionsTest.parquet"
628630 f , err := os .CreateTemp ("" , fileName )
629631 require .NoError (t , err )
632+ defer removeFileUnlessFailed (t , f )
630633
631634 writer , err := NewWriter (schemaDef , f , opt )
632635 require .NoError (t , err )
@@ -649,6 +652,16 @@ func optionsTest(t *testing.T, opt Option, testFn func(t *testing.T, reader *fil
649652 require .NoError (t , err )
650653}
651654
655+ func removeFileUnlessFailed (t * testing.T , f * os.File ) {
656+ t .Helper ()
657+ if ! t .Failed () {
658+ return
659+ }
660+ if err := os .Remove (f .Name ()); err != nil {
661+ t .Logf ("failed to remove file %s: %v" , f .Name (), err )
662+ }
663+ }
664+
652665func TestSquashTuples (t * testing.T ) {
653666 datums := []tree.Datum {
654667 tree .NewDInt (1 ),
0 commit comments