Geografic-Information-System/lib/python/pygrass/modules/interface/testsuite/test_modules.py

# -*- coding: utf-8 -*-
"""
Created on Tue Jun 24 09:43:53 2014

@author: pietro
"""
from fnmatch import fnmatch
from grass.gunittest.case import TestCase
from grass.gunittest.main import test

from grass.script.core import get_commands
from grass.pygrass.modules.interface import Module


SKIP = ["g.parser", ]


class TestModulesMeta(type):
    def __new__(mcs, name, bases, dict):

        def gen_test(cmd):
            def test(self):
                Module(cmd)
            return test

        cmds = [c for c in sorted(list(get_commands()[0]))
                if c not in SKIP and not fnmatch(c, "g.gui.*")]
        for cmd in cmds:
            test_name = "test__%s" % cmd.replace('.', '_')
            dict[test_name] = gen_test(cmd)
        return type.__new__(mcs, name, bases, dict)


class TestModules(TestCase):
    __metaclass__ = TestModulesMeta


class TestModulesPickability(TestCase):
    def test_rsun(self):
        """Test if a Module instance is pickable"""
        import pickle
        from StringIO import StringIO
        out = StringIO()
        pickle.dump(Module('r.sun'), out)
        out.close()


if __name__ == '__main__':
    test()