@@ -119,6 +119,7 @@ func TestRandomDatums(t *testing.T) {
119
119
fileName := "TestRandomDatums.parquet"
120
120
f , err := os .CreateTemp ("" , fileName )
121
121
require .NoError (t , err )
122
+ defer removeFileUnlessFailed (t , f )
122
123
123
124
schemaDef , err := NewSchema (sch .columnNames , sch .columnTypes )
124
125
require .NoError (t , err )
@@ -520,6 +521,7 @@ func TestBasicDatums(t *testing.T) {
520
521
fileName := "TestBasicDatums.parquet"
521
522
f , err := os .CreateTemp ("" , fileName )
522
523
require .NoError (t , err )
524
+ defer removeFileUnlessFailed (t , f )
523
525
524
526
schemaDef , err := NewSchema (tc .sch .columnNames , tc .sch .columnTypes )
525
527
require .NoError (t , err )
@@ -627,6 +629,7 @@ func optionsTest(t *testing.T, opt Option, testFn func(t *testing.T, reader *fil
627
629
fileName := "OptionsTest.parquet"
628
630
f , err := os .CreateTemp ("" , fileName )
629
631
require .NoError (t , err )
632
+ defer removeFileUnlessFailed (t , f )
630
633
631
634
writer , err := NewWriter (schemaDef , f , opt )
632
635
require .NoError (t , err )
@@ -649,6 +652,16 @@ func optionsTest(t *testing.T, opt Option, testFn func(t *testing.T, reader *fil
649
652
require .NoError (t , err )
650
653
}
651
654
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
+
652
665
func TestSquashTuples (t * testing.T ) {
653
666
datums := []tree.Datum {
654
667
tree .NewDInt (1 ),
0 commit comments