test_algorithm_dependencies.py 4.32 KB
Newer Older
André Anjos's avatar
André Anjos committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39
#!/usr/bin/env python
# vim: set fileencoding=utf-8 :

###############################################################################
#                                                                             #
# Copyright (c) 2016 Idiap Research Institute, http://www.idiap.ch/           #
# Contact: beat.support@idiap.ch                                              #
#                                                                             #
# This file is part of the beat.core module of the BEAT platform.             #
#                                                                             #
# Commercial License Usage                                                    #
# Licensees holding valid commercial BEAT licenses may use this file in       #
# accordance with the terms contained in a written agreement between you      #
# and Idiap. For further information contact tto@idiap.ch                     #
#                                                                             #
# Alternatively, this file may be used under the terms of the GNU Affero      #
# Public License version 3 as published by the Free Software and appearing    #
# in the file LICENSE.AGPL included in the packaging of this file.            #
# The BEAT platform is distributed in the hope that it will be useful, but    #
# WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY  #
# or FITNESS FOR A PARTICULAR PURPOSE.                                        #
#                                                                             #
# You should have received a copy of the GNU Affero Public License along      #
# with the BEAT platform. If not, see http://www.gnu.org/licenses/.           #
#                                                                             #
###############################################################################


import nose.tools

from ..algorithm import Algorithm, Storage
from ..library import Library, Storage as LibraryStorage
from ..dataformat import DataFormat, Storage as DataFormatStorage

from . import prefix, tmp_prefix
from .utils import cleanup

def copy_objects(algo, lib):

Philip ABBET's avatar
Philip ABBET committed
40 41 42
    a = Algorithm(prefix, algo)
    storage = Storage(tmp_prefix, a.name)
    a.write(storage)
André Anjos's avatar
André Anjos committed
43

Philip ABBET's avatar
Philip ABBET committed
44 45 46
    for d in a.dataformats:
        storage = DataFormatStorage(tmp_prefix, d)
        a.dataformats[d].write(storage)
André Anjos's avatar
André Anjos committed
47

Philip ABBET's avatar
Philip ABBET committed
48 49 50
    l = Library(prefix, lib)
    storage = LibraryStorage(tmp_prefix, l.name)
    l.write(storage)
André Anjos's avatar
André Anjos committed
51 52 53 54


@nose.tools.with_setup(teardown=cleanup)
def test_dependencies():
Philip ABBET's avatar
Philip ABBET committed
55 56
    name = 'user/for_dep/1'
    dep_name = 'user/dep/1'
André Anjos's avatar
André Anjos committed
57

Philip ABBET's avatar
Philip ABBET committed
58
    copy_objects(name, dep_name)
André Anjos's avatar
André Anjos committed
59

Philip ABBET's avatar
Philip ABBET committed
60 61 62 63
    a = Algorithm(tmp_prefix, name)
    assert a.valid, '\n  * %s' % '\n  * '.join(a.errors)
    nose.tools.eq_(len(a.uses), 0)
    nose.tools.eq_(len(a.libraries), 0)
André Anjos's avatar
André Anjos committed
64

Philip ABBET's avatar
Philip ABBET committed
65 66
    l_dep = Library(tmp_prefix, dep_name)
    assert l_dep.valid, '\n  * %s' % '\n  * '.join(l_dep.errors)
André Anjos's avatar
André Anjos committed
67

Philip ABBET's avatar
Philip ABBET committed
68 69 70 71 72
    # check modification
    a.uses['dep1'] = dep_name
    a.write()
    a = Algorithm(tmp_prefix, name)
    assert a.valid, '\n  * %s' % '\n  * '.join(a.errors)
André Anjos's avatar
André Anjos committed
73

Philip ABBET's avatar
Philip ABBET committed
74 75 76 77
    nose.tools.eq_(len(a.uses), 1)
    nose.tools.eq_(len(a.libraries), 1)
    nose.tools.eq_(a.uses.keys()[0], 'dep1')
    nose.tools.eq_(a.uses.values()[0], 'user/dep/1')
André Anjos's avatar
André Anjos committed
78

Philip ABBET's avatar
Philip ABBET committed
79 80 81 82 83
    a.uses = {}
    a.uses['mod1'] = dep_name
    a.write()
    a = Algorithm(tmp_prefix, name)
    assert a.valid, '\n  * %s' % '\n  * '.join(a.errors)
André Anjos's avatar
André Anjos committed
84

Philip ABBET's avatar
Philip ABBET committed
85 86 87 88
    nose.tools.eq_(len(a.uses), 1)
    nose.tools.eq_(len(a.libraries), 1)
    nose.tools.eq_(a.uses.keys()[0], 'mod1')
    nose.tools.eq_(a.uses.values()[0], 'user/dep/1')
André Anjos's avatar
André Anjos committed
89

Philip ABBET's avatar
Philip ABBET committed
90 91 92 93
    a.uses = {}
    a.write()
    a = Algorithm(tmp_prefix, name)
    assert a.valid, '\n  * %s' % '\n  * '.join(a.errors)
André Anjos's avatar
André Anjos committed
94

Philip ABBET's avatar
Philip ABBET committed
95 96
    nose.tools.eq_(len(a.uses), 0)
    nose.tools.eq_(len(a.libraries), 0)
André Anjos's avatar
André Anjos committed
97 98 99 100 101


@nose.tools.with_setup(teardown=cleanup)
def test_invalid_dependencies():

Philip ABBET's avatar
Philip ABBET committed
102 103
    name = 'user/for_dep/1'
    dep_name = 'user/invalid_dep/1'
André Anjos's avatar
André Anjos committed
104

Philip ABBET's avatar
Philip ABBET committed
105
    copy_objects(name, dep_name)
André Anjos's avatar
André Anjos committed
106

Philip ABBET's avatar
Philip ABBET committed
107 108 109 110
    a = Algorithm(tmp_prefix, name)
    assert a.valid, '\n  * %s' % '\n  * '.join(a.errors)
    nose.tools.eq_(len(a.uses), 0)
    nose.tools.eq_(len(a.libraries), 0)
André Anjos's avatar
André Anjos committed
111

Philip ABBET's avatar
Philip ABBET committed
112 113
    l_dep = Library(tmp_prefix, 'user/invalid_dep/1')
    assert l_dep.valid, '\n  * %s' % '\n  * '.join(l_dep.errors)
André Anjos's avatar
André Anjos committed
114

Philip ABBET's avatar
Philip ABBET committed
115 116 117 118 119
    a.uses['dep'] = dep_name
    a.write()
    a = Algorithm(tmp_prefix, name)
    assert not a.valid
    assert a.errors[0].find('differs from current language') != -1