112882Sspwilson2@wisc.edu# Copyright (c) 2017 Mark D. Hill and David A. Wood 212882Sspwilson2@wisc.edu# All rights reserved. 312882Sspwilson2@wisc.edu# 412882Sspwilson2@wisc.edu# Redistribution and use in source and binary forms, with or without 512882Sspwilson2@wisc.edu# modification, are permitted provided that the following conditions are 612882Sspwilson2@wisc.edu# met: redistributions of source code must retain the above copyright 712882Sspwilson2@wisc.edu# notice, this list of conditions and the following disclaimer; 812882Sspwilson2@wisc.edu# redistributions in binary form must reproduce the above copyright 912882Sspwilson2@wisc.edu# notice, this list of conditions and the following disclaimer in the 1012882Sspwilson2@wisc.edu# documentation and/or other materials provided with the distribution; 1112882Sspwilson2@wisc.edu# neither the name of the copyright holders nor the names of its 1212882Sspwilson2@wisc.edu# contributors may be used to endorse or promote products derived from 1312882Sspwilson2@wisc.edu# this software without specific prior written permission. 1412882Sspwilson2@wisc.edu# 1512882Sspwilson2@wisc.edu# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 1612882Sspwilson2@wisc.edu# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 1712882Sspwilson2@wisc.edu# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 1812882Sspwilson2@wisc.edu# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 1912882Sspwilson2@wisc.edu# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 2012882Sspwilson2@wisc.edu# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 2112882Sspwilson2@wisc.edu# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 2212882Sspwilson2@wisc.edu# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 2312882Sspwilson2@wisc.edu# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 2412882Sspwilson2@wisc.edu# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 2512882Sspwilson2@wisc.edu# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 2612882Sspwilson2@wisc.edu# 2712882Sspwilson2@wisc.edu# Authors: Sean Wilson 2812882Sspwilson2@wisc.edu 2912882Sspwilson2@wisc.edu 3012882Sspwilson2@wisc.eduimport helper 3112882Sspwilson2@wisc.eduimport runner as runner_mod 3212882Sspwilson2@wisc.edu 3312882Sspwilson2@wisc.educlass TestSuite(object): 3412882Sspwilson2@wisc.edu ''' 3512882Sspwilson2@wisc.edu An object grouping a collection of tests. It provides tags which enable 3612882Sspwilson2@wisc.edu filtering during list and run selection. All tests held in the suite must 3712882Sspwilson2@wisc.edu have a unique name. 3812882Sspwilson2@wisc.edu 3912882Sspwilson2@wisc.edu ..note:: 4012882Sspwilson2@wisc.edu The :func:`__new__` method enables collection of test cases, it must 4112882Sspwilson2@wisc.edu be called in order for test cases to be collected. 4212882Sspwilson2@wisc.edu 4312882Sspwilson2@wisc.edu ..note:: 4412882Sspwilson2@wisc.edu To reduce test definition boilerplate, the :func:`init` method is 4512882Sspwilson2@wisc.edu forwarded all `*args` and `**kwargs`. This means derived classes can 4612882Sspwilson2@wisc.edu define init without boilerplate super().__init__(*args, **kwargs). 4712882Sspwilson2@wisc.edu ''' 4812882Sspwilson2@wisc.edu runner = runner_mod.SuiteRunner 4912882Sspwilson2@wisc.edu collector = helper.InstanceCollector() 5012882Sspwilson2@wisc.edu fixtures = [] 5112882Sspwilson2@wisc.edu tests = [] 5212882Sspwilson2@wisc.edu tags = set() 5312882Sspwilson2@wisc.edu 5412882Sspwilson2@wisc.edu def __new__(klass, *args, **kwargs): 5512882Sspwilson2@wisc.edu obj = super(TestSuite, klass).__new__(klass, *args, **kwargs) 5612882Sspwilson2@wisc.edu TestSuite.collector.collect(obj) 5712882Sspwilson2@wisc.edu return obj 5812882Sspwilson2@wisc.edu 5912882Sspwilson2@wisc.edu def __init__(self, name=None, fixtures=tuple(), tests=tuple(), 6012882Sspwilson2@wisc.edu tags=tuple(), **kwargs): 6112882Sspwilson2@wisc.edu self.fixtures = self.fixtures + list(fixtures) 6212882Sspwilson2@wisc.edu self.tags = self.tags | set(tags) 6312882Sspwilson2@wisc.edu self.tests = self.tests + list(tests) 6412882Sspwilson2@wisc.edu if name is None: 6512882Sspwilson2@wisc.edu name = self.__class__.__name__ 6612882Sspwilson2@wisc.edu self.name = name 6712882Sspwilson2@wisc.edu 6812882Sspwilson2@wisc.edu def __iter__(self): 6912882Sspwilson2@wisc.edu return iter(self.tests)