A metaclass for marking test cases (including superclasses) as requiring network connection (for nosetests -A "not network" to work)

gistfile1.txt

# Author: Wes McKinney
# License: MIT

from pandas.util.testing import network

class NetworkTests(type):

    def __new__(cls, name, bases, attrs):
        for k, v in attrs.iteritems():
            if k.startswith('test_'):
                attrs[k] = network(v)

        for base in bases:
            for k, v in base.__dict__.iteritems():
                if k.startswith('test_'):
                    attrs[k] = network(v)
        return type.__new__(cls, name, bases, attrs)