plotter/tests/setup_test/test.py at 6ac648723b4d7fffbd79987c199327b7706d0404 · pshriwise/plotter · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
import filecmp
import shutil

import pytest
from PySide6 import QtGui, QtWidgets

from openmc_plotter.main_window import MainWindow, _openmcReload

@pytest.fixture
def run_in_tmpdir(tmpdir):
    orig = tmpdir.chdir()
    try:
        yield
    finally:
        orig.chdir()

def test_window(tmpdir, qtbot):
    orig = tmpdir.chdir()
    mw = MainWindow(model_path=orig)
    _openmcReload(model_path=orig)
    mw.loadGui()

    try:
        assert mw.waitForPlotIdle(60000)
        mw.saveImage(tmpdir / 'test.png')

        qtbot.addWidget(mw)
    finally:
        orig.chdir()

    filecmp.cmp(orig / 'ref.png', tmpdir / 'test.png')

    mw.close()

def test_batch_image(tmpdir, qtbot):
    orig = tmpdir.chdir()

    # move view file into tmpdir
    shutil.copy2(orig / 'test.pltvw', tmpdir)
    shutil.copy2(orig / 'test1.pltvw', tmpdir)

    _openmcReload(model_path=orig)

    mw = MainWindow(model_path=orig)
    mw.loadGui()

    try:
        mw.saveBatchImage('test.pltvw')
        qtbot.addWidget(mw)

        mw.saveBatchImage('test1.pltvw')
        qtbot.addWidget(mw)
    finally:
        orig.chdir()

    filecmp.cmp(orig / 'ref.png', tmpdir / 'test.png')
    filecmp.cmp(orig / 'ref1.png', tmpdir / 'test1.png')

    mw.close()

def test_copy_image_to_clipboard(tmpdir, monkeypatch, qtbot):
    orig = tmpdir.chdir()
    QtWidgets.QApplication.instance() or QtWidgets.QApplication([])
    mw = MainWindow(model_path=orig)
    _openmcReload(model_path=orig)
    mw.loadGui()
    qtbot.addWidget(mw)
    mw.show()

    class FakeClipboard:
        def __init__(self):
            self.image = None

        def setImage(self, image):
            self.image = image

    fake_clipboard = FakeClipboard()
    monkeypatch.setattr(QtGui.QGuiApplication,
                        'clipboard',
                        staticmethod(lambda: fake_clipboard))

    try:
        assert mw.waitForPlotIdle(60000)
        mw.model.currentView.domainVisible = False
        mw.plotIm.updatePixmap()
        assert mw.copyImageToClipboard()
    finally:
        orig.chdir()

    assert fake_clipboard.image is not None
    assert not fake_clipboard.image.isNull()
    assert fake_clipboard.image.hasAlphaChannel()

    canvas_width, canvas_height = mw.plotIm.get_width_height()
    expected_width = round(
        min(mw.frame.viewport().width(), mw.plotIm.width())
        * canvas_width
        / mw.plotIm.width()
    )
    expected_height = round(
        min(mw.frame.viewport().height(), mw.plotIm.height())
        * canvas_height
        / mw.plotIm.height()
    )
    assert fake_clipboard.image.width() == pytest.approx(expected_width, abs=1)
    assert fake_clipboard.image.height() == pytest.approx(expected_height, abs=1)

    center = fake_clipboard.image.pixelColor(fake_clipboard.image.width() // 2,
                                             fake_clipboard.image.height() // 2)
    assert center.alpha() == 0

    mw.close()