mirror of
https://github.com/pyapp-kit/superqt.git
synced 2026-01-06 12:21:17 +01:00
8457563f49
* Implement throttling of methods * style: [pre-commit.ci] auto fixes [...] * fix line length * chek if object instance is Qt object * handle `self._name` being None or empty string * fix throttling method * handle staticmethod * use descriptor * try fix staticmethods * move descriptor to a separate class * move __set_name__ * simplify code and restore timer information * inspire tlamber suggestions * clean code * add weakref dict as fallback --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
134 lines
2.7 KiB
Python
134 lines
2.7 KiB
Python
from unittest.mock import Mock
|
|
|
|
import pytest
|
|
from qtpy.QtCore import QObject, Signal
|
|
|
|
from superqt.utils import qdebounced, qthrottled
|
|
from superqt.utils._throttler import ThrottledCallable
|
|
|
|
|
|
def test_debounced(qtbot):
|
|
mock1 = Mock()
|
|
mock2 = Mock()
|
|
|
|
@qdebounced(timeout=5)
|
|
def f1() -> str:
|
|
mock1()
|
|
|
|
def f2() -> str:
|
|
mock2()
|
|
|
|
for _ in range(10):
|
|
f1()
|
|
f2()
|
|
|
|
qtbot.wait(5)
|
|
mock1.assert_called_once()
|
|
assert mock2.call_count == 10
|
|
|
|
|
|
def test_debouncer_method(qtbot):
|
|
class A(QObject):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.count = 0
|
|
|
|
def callback(self):
|
|
self.count += 1
|
|
|
|
a = A()
|
|
assert all(not isinstance(x, ThrottledCallable) for x in a.children())
|
|
b = qdebounced(a.callback, timeout=4)
|
|
assert any(isinstance(x, ThrottledCallable) for x in a.children())
|
|
for _ in range(10):
|
|
b()
|
|
|
|
qtbot.wait(5)
|
|
|
|
assert a.count == 1
|
|
|
|
|
|
def test_debouncer_method_definition(qtbot):
|
|
mock1 = Mock()
|
|
mock2 = Mock()
|
|
|
|
class A(QObject):
|
|
def __init__(self):
|
|
super().__init__()
|
|
self.count = 0
|
|
|
|
@qdebounced(timeout=4)
|
|
def callback(self):
|
|
self.count += 1
|
|
|
|
@qdebounced(timeout=4)
|
|
@staticmethod
|
|
def call1():
|
|
mock1()
|
|
|
|
@staticmethod
|
|
@qdebounced(timeout=4)
|
|
def call2():
|
|
mock2()
|
|
|
|
a = A()
|
|
assert all(not isinstance(x, ThrottledCallable) for x in a.children())
|
|
for _ in range(10):
|
|
a.callback(1)
|
|
A.call1(34)
|
|
a.call1(22)
|
|
a.call2(22)
|
|
A.call2(32)
|
|
|
|
qtbot.wait(5)
|
|
|
|
assert a.count == 1
|
|
mock1.assert_called_once()
|
|
mock2.assert_called_once()
|
|
|
|
|
|
def test_throttled(qtbot):
|
|
mock1 = Mock()
|
|
mock2 = Mock()
|
|
|
|
@qthrottled(timeout=5)
|
|
def f1() -> str:
|
|
mock1()
|
|
|
|
def f2() -> str:
|
|
mock2()
|
|
|
|
for _ in range(10):
|
|
f1()
|
|
f2()
|
|
|
|
qtbot.wait(5)
|
|
assert mock1.call_count == 2
|
|
assert mock2.call_count == 10
|
|
|
|
|
|
@pytest.mark.parametrize("deco", [qthrottled, qdebounced])
|
|
def test_ensure_throttled_sig_inspection(deco, qtbot):
|
|
mock = Mock()
|
|
|
|
class Emitter(QObject):
|
|
sig = Signal(int, int, int)
|
|
|
|
@deco
|
|
def func(a: int, b: int):
|
|
"""docstring"""
|
|
mock(a, b)
|
|
|
|
obj = Emitter()
|
|
obj.sig.connect(func)
|
|
|
|
# this is the crux of the test...
|
|
# we emit 3 args, but the function only takes 2
|
|
# this should normally work fine in Qt.
|
|
# testing here that the decorator doesn't break it.
|
|
with qtbot.waitSignal(func.triggered, timeout=1000):
|
|
obj.sig.emit(1, 2, 3)
|
|
mock.assert_called_once_with(1, 2)
|
|
assert func.__doc__ == "docstring"
|
|
assert func.__name__ == "func"
|