diff --git a/notebook_test.py b/notebook_test.py index eb968e7e..2ea5675f 100644 --- a/notebook_test.py +++ b/notebook_test.py @@ -5,7 +5,7 @@ from utils import FileHandler from utils.enums import Venue - +import itertools def parse_args(): """ @@ -67,8 +67,8 @@ def run(): if path.exists(inputFile): venue = Venue.from_str(environment) collections = FileHandler.get_file_content_list_per_line(inputFile) - for collection in collections: - + # limit number of collections tested to 4 + for collection in itertools.islice(collections, 3): if "POCLOUD" not in collection and venue == "uat": continue