File: test_star.py

package info (click to toggle)
fpdf2 2.8.4-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 53,860 kB
  • sloc: python: 39,487; sh: 133; makefile: 12
file content (101 lines) | stat: -rw-r--r-- 4,317 bytes parent folder | download | duplicates (2)
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
from pathlib import Path

import fpdf
from test.conftest import assert_pdf_equal

import pytest


HERE = Path(__file__).resolve().parent


def test_star(tmp_path):
    pdf = fpdf.FPDF()
    pdf.add_page()

    # no fill with line test
    y = 20
    pdf.star(x=15, y=y, r_in=5, r_out=15, corners=3, style="D")
    pdf.star(x=45, y=y, r_in=5, r_out=15, corners=4, style="D")
    pdf.star(x=75, y=y, r_in=5, r_out=15, corners=5, style="D")
    pdf.star(x=105, y=y, r_in=5, r_out=15, corners=6, style="D")
    pdf.star(x=135, y=y, r_in=5, r_out=15, corners=7, style="D")
    pdf.star(x=165, y=y, r_in=5, r_out=15, corners=8, style="D")
    pdf.star(x=195, y=y, r_in=5, r_out=15, corners=9, style="D")

    # fill and color test
    y += 40
    pdf.set_fill_color(r=134, g=200, b=15)
    pdf.star(x=15, y=y, r_in=5, r_out=15, corners=3, style="DF")
    pdf.star(x=45, y=y, r_in=5, r_out=15, corners=4, style="DF")
    pdf.star(x=75, y=y, r_in=5, r_out=15, corners=5, style="DF")
    pdf.star(x=105, y=y, r_in=5, r_out=15, corners=6, style="DF")
    pdf.star(x=135, y=y, r_in=5, r_out=15, corners=7, style="DF")
    pdf.star(x=165, y=y, r_in=5, r_out=15, corners=8, style="DF")
    pdf.star(x=195, y=y, r_in=5, r_out=15, corners=9, style="DF")

    # draw color test
    y += 40
    pdf.set_fill_color(r=0, g=0, b=0)
    pdf.set_draw_color(r=204, g=0, b=204)
    pdf.star(x=15, y=y, r_in=5, r_out=15, corners=3, style="D")
    pdf.star(x=45, y=y, r_in=5, r_out=15, corners=4, style="D")
    pdf.star(x=75, y=y, r_in=5, r_out=15, corners=5, style="D")
    pdf.star(x=105, y=y, r_in=5, r_out=15, corners=6, style="D")
    pdf.star(x=135, y=y, r_in=5, r_out=15, corners=7, style="D")
    pdf.star(x=165, y=y, r_in=5, r_out=15, corners=8, style="D")
    pdf.star(x=195, y=y, r_in=5, r_out=15, corners=9, style="D")

    # line width test
    y += 40
    pdf.set_line_width(1)
    pdf.star(x=15, y=y, r_in=5, r_out=15, corners=3, style="D")
    pdf.star(x=45, y=y, r_in=5, r_out=15, corners=4, style="D")
    pdf.star(x=75, y=y, r_in=5, r_out=15, corners=5, style="D")
    pdf.star(x=105, y=y, r_in=5, r_out=15, corners=6, style="D")
    pdf.star(x=135, y=y, r_in=5, r_out=15, corners=7, style="D")
    pdf.star(x=165, y=y, r_in=5, r_out=15, corners=8, style="D")
    pdf.star(x=195, y=y, r_in=5, r_out=15, corners=9, style="D")

    # line color and fill color
    y += 40
    pdf.set_fill_color(r=3, g=190, b=252)
    pdf.star(x=15, y=y, r_in=5, r_out=15, corners=3, style="DF")
    pdf.star(x=45, y=y, r_in=5, r_out=15, corners=4, style="DF")
    pdf.star(x=75, y=y, r_in=5, r_out=15, corners=5, style="DF")
    pdf.star(x=105, y=y, r_in=5, r_out=15, corners=6, style="DF")
    pdf.star(x=135, y=y, r_in=5, r_out=15, corners=7, style="DF")
    pdf.star(x=165, y=y, r_in=5, r_out=15, corners=8, style="DF")
    pdf.star(x=195, y=y, r_in=5, r_out=15, corners=9, style="DF")

    # fill only
    y += 40
    pdf.set_draw_color(r=0, g=0, b=255)
    pdf.star(x=15, y=y, r_in=5, r_out=15, corners=3, style="F")
    pdf.star(x=45, y=y, r_in=5, r_out=15, corners=4, style="F")
    pdf.star(x=75, y=y, r_in=5, r_out=15, corners=5, style="F")
    pdf.star(x=105, y=y, r_in=5, r_out=15, corners=6, style="F")
    pdf.star(x=135, y=y, r_in=5, r_out=15, corners=7, style="F")
    pdf.star(x=165, y=y, r_in=5, r_out=15, corners=8, style="F")
    pdf.star(x=195, y=y, r_in=5, r_out=15, corners=9, style="F")

    # rotation test
    y += 40
    pdf.set_draw_color(r=0, g=0, b=255)
    pdf.star(x=15, y=y, r_in=5, r_out=15, corners=3, rotate_degrees=0, style="DF")
    pdf.star(x=45, y=y, r_in=5, r_out=15, corners=4, rotate_degrees=35, style="DF")
    pdf.star(x=75, y=y, r_in=5, r_out=15, corners=5, rotate_degrees=45, style="DF")
    pdf.star(x=105, y=y, r_in=5, r_out=15, corners=6, rotate_degrees=200, style="DF")
    pdf.star(x=135, y=y, r_in=5, r_out=15, corners=7, rotate_degrees=13, style="DF")
    pdf.star(x=165, y=y, r_in=5, r_out=15, corners=8, rotate_degrees=22.5, style="DF")
    pdf.star(x=195, y=y, r_in=5, r_out=15, corners=9, rotate_degrees=77.3, style="DF")

    assert_pdf_equal(pdf, HERE / "regular_star.pdf", tmp_path)


def test_star_invalid_style():
    pdf = fpdf.FPDF()
    pdf.add_page()

    with pytest.raises(ValueError):
        pdf.star(x=15, y=15, r_in=5, r_out=15, corners=3, rotate_degrees=0, style="N")