Skip to content

Commit 5255625

Browse files
mariuzkagithub-actions[bot]
authored andcommitted
auto-commit: fixes by ruff
1 parent 0270e2a commit 5255625

2 files changed

Lines changed: 25 additions & 17 deletions

File tree

tests/test_gui/test_default_buttons.py

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1,29 +1,30 @@
11
from mesa.examples.basic.schelling.model import Schelling
22
import mesarcade as mesar
33

4+
45
def test_play_button():
56
canvas = mesar.Canvas(
67
model_class=Schelling,
78
_visible=False,
89
)
910
canvas._setup()
10-
11+
1112
for _ in range(10):
1213
canvas.renderer.on_update(None)
1314

1415
assert canvas.renderer.play == False
1516
assert canvas.renderer.tick == 0
1617

1718
canvas.renderer.default_buttons.play_button.on_click(None)
18-
19+
1920
for _ in range(10):
2021
canvas.renderer.on_update(None)
21-
22+
2223
assert canvas.renderer.play == True
2324
assert canvas.renderer.tick == 10
2425

2526
canvas.renderer.default_buttons.play_button.on_click(None)
26-
27+
2728
assert canvas.renderer.play == False
2829
assert canvas.renderer.tick == 10
2930

@@ -37,12 +38,12 @@ def test_reset_button():
3738
)
3839
canvas._setup()
3940
canvas.renderer.play = True
40-
41+
4142
assert canvas.renderer.tick == 0
4243

4344
for _ in range(10):
4445
canvas.renderer.on_update(None)
45-
46+
4647
assert canvas.renderer.tick == 10
4748

4849
canvas.renderer.default_buttons.reset_button.on_click(None)
@@ -66,4 +67,4 @@ def test_step_button():
6667

6768
assert canvas.renderer.tick == 15
6869

69-
canvas.window.close()
70+
canvas.window.close()

tests/test_gui/test_num_controllers.py

Lines changed: 17 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -2,15 +2,21 @@
22
import mesarcade as mesar
33
from dataclasses import dataclass
44

5+
56
@dataclass
67
class Event:
78
new_value: float
89

910

1011
def create_canvas():
1112
population_size_controller = mesar.NumController(
12-
"population_size", 100, 25, 1000, 25, "n birds",
13-
)
13+
"population_size",
14+
100,
15+
25,
16+
1000,
17+
25,
18+
"n birds",
19+
)
1420
canvas = mesar.Canvas(
1521
model_class=BoidFlockers,
1622
controllers=[population_size_controller],
@@ -19,11 +25,12 @@ def create_canvas():
1925
canvas._setup()
2026
return canvas
2127

28+
2229
def test_num_controller_buttons():
2330
canvas = create_canvas()
24-
31+
2532
controller = canvas.renderer.controllers[0]
26-
33+
2734
# test basic setup
2835
assert controller.parameter_name == "population_size"
2936
assert controller.label == "n birds"
@@ -42,23 +49,23 @@ def test_num_controller_buttons():
4249
# test maximum
4350
for _ in range(500):
4451
controller.buttons.increase_button.on_click(None)
45-
52+
4653
assert controller.parameter_value == 1000
4754
assert canvas.renderer.parameter_dict["population_size"] == 1000
4855

4956
# test minimum
5057
for _ in range(500):
5158
controller.buttons.decrease_button.on_click(None)
52-
59+
5360
assert controller.parameter_value == 25
5461
assert canvas.renderer.parameter_dict["population_size"] == 25
55-
62+
5663
canvas.window.close()
5764

5865

5966
def test_num_controller_slider():
6067
canvas = create_canvas()
61-
68+
6269
controller = canvas.renderer.controllers[0]
6370

6471
assert controller.parameter_value == 100
@@ -72,7 +79,7 @@ def test_num_controller_slider():
7279

7380
def test_increase_and_reset():
7481
canvas = create_canvas()
75-
82+
7683
controller = canvas.renderer.controllers[0]
7784

7885
assert len(canvas.renderer.model.agents) == 100
@@ -85,4 +92,4 @@ def test_increase_and_reset():
8592

8693
assert len(canvas.renderer.model.agents) == 125
8794

88-
canvas.window.close()
95+
canvas.window.close()

0 commit comments

Comments
 (0)