close h2o session in test

AlouiAmine · Nov 13, 2018 · 0020fba · 0020fba
1 parent 3fc7e45
commit 0020fba
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/test_h2o.py b/tests/test_h2o.py
@@ -3,6 +3,9 @@
 import h2o
 
 class TestH2o(unittest.TestCase):
+    def tearDown(self):
+        h2o.cluster().shutdown(False)
+
     def test_init_read(self):
         h2o.init()
         train = h2o.import_file("/input/tests/data/train.csv", destination_frame="train")