tsi-org/tango

1# Copyright 2023 The HuggingFace Team. All rights reserved.2#3# Licensed under the Apache License, Version 2.0 (the "License");4# you may not use this file except in compliance with the License.5# You may obtain a copy of the License at6#7#     http://www.apache.org/licenses/LICENSE-2.08#9# Unless required by applicable law or agreed to in writing, software10# distributed under the License is distributed on an "AS IS" BASIS,11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.12# See the License for the specific language governing permissions and13# limitations under the License.14 15# tests directory-specific settings - this file is run automatically16# by pytest before any tests are run17 18import sys19import warnings20from os.path import abspath, dirname, join21 22 23# allow having multiple repository checkouts and not needing to remember to rerun24# 'pip install -e .[dev]' when switching between checkouts and running tests.25git_repo_path = abspath(join(dirname(dirname(dirname(__file__))), "src"))26sys.path.insert(1, git_repo_path)27 28 29# silence FutureWarning warnings in tests since often we can't act on them until30# they become normal warnings - i.e. the tests still need to test the current functionality31warnings.simplefilter(action="ignore", category=FutureWarning)32 33 34def pytest_addoption(parser):35    from diffusers.utils.testing_utils import pytest_addoption_shared36 37    pytest_addoption_shared(parser)38 39 40def pytest_terminal_summary(terminalreporter):41    from diffusers.utils.testing_utils import pytest_terminal_summary_main42 43    make_reports = terminalreporter.config.getoption("--make-reports")44    if make_reports:45        pytest_terminal_summary_main(terminalreporter, id=make_reports)46