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