Add random_walk with tests

200 steps of equal length, each in a random heading of 0, 90, 180 or 270 degrees and
a random RGB color, with a thick line and the fastest speed.

Task: MIL-004#1
This commit is contained in:
2026-10-07 23:59:10 +08:00
parent c695c47929
commit 949d2edb63
5 changed files with 183 additions and 2 deletions
+5 -2
View File
@@ -4,13 +4,15 @@
100 Days of Code.
The challenges so far are @ref turtle_challenges.square.draw_square,
@ref turtle_challenges.dashed_line.draw_dashed_line and
@ref turtle_challenges.shapes.draw_shape. Each takes the turtle as its
@ref turtle_challenges.dashed_line.draw_dashed_line,
@ref turtle_challenges.shapes.draw_shape and
@ref turtle_challenges.random_walk.random_walk. Each takes the turtle as its
first argument, so it can be tested without a window.
"""
from turtle_challenges.colors import random_color, random_palette_color
from turtle_challenges.dashed_line import draw_dashed_line
from turtle_challenges.random_walk import random_walk
from turtle_challenges.shapes import draw_shape, draw_shapes
from turtle_challenges.square import draw_square
@@ -21,4 +23,5 @@ __all__ = [
"draw_square",
"random_color",
"random_palette_color",
"random_walk",
]
+15
View File
@@ -26,6 +26,9 @@ FULL_TURN_DEGREES = 360
## Degrees in a right angle.
RIGHT_ANGLE_DEGREES = 90
## Turtle speed setting for "fastest": no animation between moves.
FASTEST_SPEED = 0
## Challenge 1: number of sides of a square.
SQUARE_SIDES = 4
@@ -62,6 +65,18 @@ COLOR_PALETTE: tuple[str, ...] = (
"SeaGreen",
)
## Challenge 4: number of steps of the random walk.
WALK_STEPS = 200
## Challenge 4: distance of one step, in turtle units.
WALK_STEP_DISTANCE = 30
## Challenge 4: thickness of the line, in pixels.
WALK_PEN_SIZE = 10
## Challenge 4: headings of a step: east, north, west and south, in degrees.
WALK_HEADINGS: tuple[int, ...] = (0, 90, 180, 270)
## Command line name of challenge 1.
CHALLENGE_SQUARE = "square"
+49
View File
@@ -0,0 +1,49 @@
"""!
@file random_walk.py
@brief Turtle Challenge 4: generate a random walk.
"""
import random
from turtle_challenges.colors import random_color
from turtle_challenges.constants import (
FASTEST_SPEED,
WALK_HEADINGS,
WALK_PEN_SIZE,
WALK_STEP_DISTANCE,
WALK_STEPS,
)
from turtle_challenges.pen import Pen
def random_walk(
pen: Pen,
steps: int = WALK_STEPS,
distance: float = WALK_STEP_DISTANCE,
rng: random.Random | None = None,
) -> None:
"""!
@brief Walk in random directions, changing color at every step.
Each step faces east, north, west or south at random and moves the same
distance in a new random RGB color.
@param pen The turtle that draws.
@param steps Number of steps to take.
@param distance Distance of one step, in turtle units.
@param rng Random generator for headings and colors; a shared one is used
if omitted.
@exception ValueError If @p steps is negative, or @p distance is not
positive.
"""
if steps < 0:
raise ValueError(f"steps must not be negative, got {steps}")
if distance <= 0:
raise ValueError(f"distance must be positive, got {distance}")
generator = rng if rng is not None else random.Random()
pen.pensize(WALK_PEN_SIZE)
pen.speed(FASTEST_SPEED)
for _ in range(steps):
pen.color(random_color(generator))
pen.setheading(generator.choice(WALK_HEADINGS))
pen.forward(distance)