test_collector.py

 
import basf2
from zmq_daq import test_support
from unittest import main
 
 
class CollectorTestCase(test_support.BaseCollectorTestCase):
    """Test case"""
    
    final_collector = False
 
 
class FinalCollectorTestCase(test_support.BaseCollectorTestCase):
    """Test case"""
    
    final_collector = True
 
 
if __name__ == '__main__':
    
    number_of_failures = 0
 
    for i in range(test_support.ZMQ_TEST_FOR_LOOPS):
        try:
            main(exit=False)
        except AssertionError:
            number_of_failures += 1
 
    
    message = f'Number of failed for loops: {number_of_failures}/{test_support.ZMQ_TEST_FOR_LOOPS}'
    if number_of_failures <= test_support.ZMQ_TEST_MAX_FAILURES:
        basf2.B2INFO(message)
    else:
        basf2.B2FATAL(message)