address feedback

Author: BryanCutler

tensorflow_io/arrow/python/kernel_tests/arrow_test.py

@@ -98,9 +98,7 @@ def run_test_case(self, dataset, case_data):
     next_element = iterator.get_next()
 
     def is_float(dtype):
-      return dtype == dtypes.float16 or \
-             dtype == dtypes.float32 or \
-             dtype == dtypes.float64
+      return dtype in [dtypes.float16, dtypes.float32, dtypes.float64]
 
     with self.test_session() as sess:
       for row in range(len(case_data.data[0])):
@@ -200,9 +198,9 @@ def testArrowFeatherDataset(self):
     batch = self.make_record_batch(test_data)
     df = batch.to_pandas()
 
-    f = tempfile.NamedTemporaryFile(delete=False)
-    write_feather(df, f)
-    f.close()
+    # Create a tempfile that is deleted after tests run
+    with tempfile.NamedTemporaryFile(delete=False) as f:
+      write_feather(df, f)
 
     # test single file
     dataset = arrow_dataset_ops.ArrowFeatherDataset(