diff --git a/tests/test_ui_mpldrawer.py b/tests/test_ui_mpldrawer.py index 619042d226..ec48c9da89 100644 --- a/tests/test_ui_mpldrawer.py +++ b/tests/test_ui_mpldrawer.py @@ -1,10 +1,12 @@ """Tests for Qibo matplotlib drawer""" -from PIL import Image -import numpy as np -import pytest from pathlib import Path + import matplotlib.pyplot +import numpy as np +import pytest +from PIL import Image + from qibo import Circuit, callbacks, gates from qibo.models import QFT from qibo.ui.drawer_utils import FusedEndGateBarrier, FusedStartGateBarrier @@ -19,6 +21,7 @@ BASEPATH = str(Path(__file__).parent / "test_ui_images") + # convert an image to RGBA numpy array def img2array(img_path): img = Image.open(img_path)