test_algorithm_dependencies.py 4.21 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 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119
#!/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):

  a = Algorithm(prefix, algo)
  storage = Storage(tmp_prefix, a.name)
  a.write(storage)

  for d in a.dataformats:
    storage = DataFormatStorage(tmp_prefix, d)
    a.dataformats[d].write(storage)

  l = Library(prefix, lib)
  storage = LibraryStorage(tmp_prefix, l.name)
  l.write(storage)


@nose.tools.with_setup(teardown=cleanup)
def test_dependencies():
  name = 'user/for_dep/1'
  dep_name = 'user/dep/1'

  copy_objects(name, dep_name)

  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)

  l_dep = Library(tmp_prefix, dep_name)
  assert l_dep.valid, '\n  * %s' % '\n  * '.join(l_dep.errors)

  # check modification
  a.uses['dep1'] = dep_name
  a.write()
  a = Algorithm(tmp_prefix, name)
  assert a.valid, '\n  * %s' % '\n  * '.join(a.errors)

  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')

  a.uses = {}
  a.uses['mod1'] = dep_name
  a.write()
  a = Algorithm(tmp_prefix, name)
  assert a.valid, '\n  * %s' % '\n  * '.join(a.errors)

  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')

  a.uses = {}
  a.write()
  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)


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

  name = 'user/for_dep/1'
  dep_name = 'user/invalid_dep/1'

  copy_objects(name, dep_name)

  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)

  l_dep = Library(tmp_prefix, 'user/invalid_dep/1')
  assert l_dep.valid, '\n  * %s' % '\n  * '.join(l_dep.errors)

  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