Merge branch 'imagetests' into 'master'

Add image tests

See merge request !99

.gitlab-ci.yml

@@ -7,7 +7,7 @@ before_script:
   - apt-get install --yes build-essential cmake libfmt-dev pybind11-dev graphviz python3-pyqt5 xvfb xdg-utils lcov
   - python -m pip install --upgrade pip
   - python --version
-  - pip install pytest pytest-cov sphinx furo numpydoc pytest-xvfb pytest-qt setuptools_scm sphinx-qt-documentation pytest-xdist
+  - pip install pytest pytest-cov sphinx furo numpydoc pytest-xvfb pytest-qt setuptools_scm sphinx-qt-documentation pytest-xdist pytest-mpl
   # - export CXXFLAGS='--coverage'
   - pip install -ve .
   # Move file, but should be handled by installation
@@ -17,7 +17,7 @@ before_script:
 .run-test:
   stage: test
   script:
-    - pytest --cov=b_asic --cov-report=xml:cov.xml --cov-report=term --color=yes test
+    - pytest --cov=b_asic --cov-report=xml:cov.xml --cov-report=term --color=yes test --mpl
     # - lcov --capture --directory . --output-file coverage.info
     # - lcov --output-file coverage.info --extract coverage.info $PWD/src/'*' $PWD/b_asic/'*'
     # - lcov --list coverage.info

b_asic/schedule.py

@@ -563,9 +563,12 @@ class Schedule:
         )
 
     def plot_schedule(self) -> None:
+        self._get_figure().show()
+
+    def _get_figure(self):
         fig, ax = plt.subplots()
         self._plot_schedule(ax)
-        fig.show()
+        return fig
 
     def _repr_svg_(self):
         fig, ax = plt.subplots()

test/test_schedule.py

@@ -467,3 +467,9 @@ class TestProcesses:
     def test__get_memory_variables_list(self, secondorder_iir_schedule):
         mvl = secondorder_iir_schedule._get_memory_variables_list()
         assert len(mvl) == 12
+
+
+class TestFigureGeneration:
+    @pytest.mark.mpl_image_compare(remove_text=True, style='mpl20')
+    def test__get_figure_no_execution_times(self, secondorder_iir_schedule):
+        return secondorder_iir_schedule._get_figure()