|
- # Copyright (c) 2021-2022, InterDigital Communications, Inc
- # All rights reserved.
-
- # Redistribution and use in source and binary forms, with or without
- # modification, are permitted (subject to the limitations in the disclaimer
- # below) provided that the following conditions are met:
-
- # * Redistributions of source code must retain the above copyright notice,
- # this list of conditions and the following disclaimer.
- # * Redistributions in binary form must reproduce the above copyright notice,
- # this list of conditions and the following disclaimer in the documentation
- # and/or other materials provided with the distribution.
- # * Neither the name of InterDigital Communications, Inc nor the names of its
- # contributors may be used to endorse or promote products derived from this
- # software without specific prior written permission.
-
- # NO EXPRESS OR IMPLIED LICENSES TO ANY PARTY'S PATENT RIGHTS ARE GRANTED BY
- # THIS LICENSE. THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND
- # CONTRIBUTORS "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT
- # NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A
- # PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR
- # CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
- # EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
- # PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS;
- # OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY,
- # WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR
- # OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF
- # ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
-
- import importlib
- import os
-
- import matplotlib
- import pytest
-
- matplotlib.use("Agg")
-
- plot = importlib.import_module("compressai.utils.plot.__main__")
-
-
- @pytest.mark.parametrize("metric", ("psnr", "ms-ssim"))
- @pytest.mark.parametrize("backend", ("matplotlib", "plotly"))
- def test_plot(metric, backend):
- here = os.path.dirname(__file__)
- filepath = os.path.join(here, "expected/eval_0_bmshj2018-factorized_mse_1.json")
- cmd = [
- "-f",
- filepath,
- "--title",
- "myplot",
- "--metric",
- metric,
- "--backend",
- backend,
- ]
- plot.main(cmd)
|