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