From e3b776756205cb6ed3235eb5ee73680867ad2e87 Mon Sep 17 00:00:00 2001
From: Andre Anjos <andre.dos.anjos@gmail.com>
Date: Thu, 15 Feb 2018 14:48:52 +0100
Subject: [PATCH] Makes sure we can load/save files with arbitrary extensions

---
 bob/learn/libsvm/test_machine.py | 13 +++++++++++--
 1 file changed, 11 insertions(+), 2 deletions(-)

diff --git a/bob/learn/libsvm/test_machine.py b/bob/learn/libsvm/test_machine.py
index 468e6a3..ee0f592 100644
--- a/bob/learn/libsvm/test_machine.py
+++ b/bob/learn/libsvm/test_machine.py
@@ -101,10 +101,10 @@ def test_can_save():
 
   os.unlink(tmp)
 
-def test_can_save_hdf5():
+def run_for_extension(ext):
 
   machine = Machine(HEART_MACHINE)
-  tmp = tempname('.hdf5')
+  tmp = tempname(ext)
   machine.save(bob.io.base.HDF5File(tmp, 'w'))
   del machine
 
@@ -123,6 +123,15 @@ def test_can_save_hdf5():
 
   os.unlink(tmp)
 
+def test_can_save_arbitrary():
+  run_for_extension('.arbitrary')
+
+def test_can_save_h5():
+  run_for_extension('.h5')
+
+def test_can_save_hdf5():
+  run_for_extension('.hdf5')
+
 def test_data_loading():
 
   #tests if I can load data in libsvm format using SVMFile
-- 
GitLab