make_test_stubs.py

from __future__ import print_function

import sys, os
from pymol import cmd

def make_test_stubs(module, out=sys.stdout):
    '''
    Utility function to generate a test class for a PyMOL module,
    filled with skipped stub methods.

    Example:

    PyMOL> run make_test_stubs.py
    PyMOL> make_test_stubs pymol.fitting, /tmp/fitting.py
    '''
    if isinstance(module, str):
        __import__(module)
        module = sys.modules[module]

    if isinstance(out, str):
        out = open(out, 'w')

    print('''
from pymol import cmd, testing, stored

class Test%s(testing.PyMOLTestCase):
''' % (module.__name__.split('.')[-1].capitalize()), file=out)

    type_func = type(cmd.align)
    ns = vars(module)
    for name in sorted(ns):
        value = ns[name]
        if isinstance(value, type_func) and \
                value.__module__ == module.__name__ and \
                value == getattr(cmd, name, None):
            nameCap = name.replace('_', ' ').title().replace(' ', '')
            print('    def test%s(self):' % nameCap, file=out)
            print('        cmd.%s' % value.__name__, file=out)
            print('        self.skipTest("TODO")', file=out)
            print('', file=out)

cmd.extend('make_test_stubs', make_test_stubs)