Blame - Lib/test/test_calendar.py - platform/external/python/cpython3

2001-12-12 05:38:08 +0000

[diff] [blame]

import calendar

import unittest

Benjamin Peterson

2008-05-20 21:35:26 +0000

[diff] [blame]

4

from test import support

Berker Peksag

ce64391

2015-05-06 06:33:17 +0300

[diff] [blame]

5

from test.support.script_helper import assert_python_ok, assert_python_failure

Alexander Belopolsky

43ca710

2010-06-14 17:32:03 +0000

[diff] [blame]

6

import time

Georg Brandl

2010-10-19 18:54:25 +0000

[diff] [blame]

7

import locale

R David Murray

2012-05-29 12:55:05 -0400

[diff] [blame]

8

import sys

Ezio Melotti

85710a4

2012-09-21 17:26:35 +0300

[diff] [blame]

9

import datetime

Serhiy Storchaka

2015-02-05 15:14:35 +0200

[diff] [blame]

10

import os

R David Murray

2012-05-29 12:55:05 -0400

[diff] [blame]

11

Alexander Belopolsky

2017-10-26 15:34:11 -0400

[diff] [blame]

12

# From https://en.wikipedia.org/wiki/Leap_year_starting_on_Saturday

13

result_0_02_text = """\

February 0

Mo Tu We Th Fr Sa Su

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

"""

result_0_text = """\

0

January February March

27

Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su

28

1 2 1 2 3 4 5 6 1 2 3 4 5

29

3 4 5 6 7 8 9 7 8 9 10 11 12 13 6 7 8 9 10 11 12

30

10 11 12 13 14 15 16 14 15 16 17 18 19 20 13 14 15 16 17 18 19

31

17 18 19 20 21 22 23 21 22 23 24 25 26 27 20 21 22 23 24 25 26

32

24 25 26 27 28 29 30 28 29 27 28 29 30 31

31

April May June

Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su

37

1 2 1 2 3 4 5 6 7 1 2 3 4

38

3 4 5 6 7 8 9 8 9 10 11 12 13 14 5 6 7 8 9 10 11

39

10 11 12 13 14 15 16 15 16 17 18 19 20 21 12 13 14 15 16 17 18

40

17 18 19 20 21 22 23 22 23 24 25 26 27 28 19 20 21 22 23 24 25

41

24 25 26 27 28 29 30 29 30 31 26 27 28 29 30

42

43

July August September

44

Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su

45

1 2 1 2 3 4 5 6 1 2 3

46

3 4 5 6 7 8 9 7 8 9 10 11 12 13 4 5 6 7 8 9 10

47

10 11 12 13 14 15 16 14 15 16 17 18 19 20 11 12 13 14 15 16 17

48

17 18 19 20 21 22 23 21 22 23 24 25 26 27 18 19 20 21 22 23 24

49

24 25 26 27 28 29 30 28 29 30 31 25 26 27 28 29 30

50

31

51

52

October November December

53

Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su

54

1 1 2 3 4 5 1 2 3

55

2 3 4 5 6 7 8 6 7 8 9 10 11 12 4 5 6 7 8 9 10

56

9 10 11 12 13 14 15 13 14 15 16 17 18 19 11 12 13 14 15 16 17

57

16 17 18 19 20 21 22 20 21 22 23 24 25 26 18 19 20 21 22 23 24

58

23 24 25 26 27 28 29 27 28 29 30 25 26 27 28 29 30 31

30 31

"""

Serhiy Storchaka

2015-02-05 15:14:35 +0200

[diff] [blame]

62

result_2004_01_text = """\

R David Murray

2012-05-29 12:55:05 -0400

[diff] [blame]

January 2004

Mo Tu We Th Fr Sa Su

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

"""

Fred Drake

2001-12-12 05:38:08 +0000

[diff] [blame]

71

Serhiy Storchaka

2015-02-05 15:14:35 +0200

[diff] [blame]

72

result_2004_text = """\

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

73

2004

74

75

January February March

76

Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su

77

1 2 3 4 1 1 2 3 4 5 6 7

78

5 6 7 8 9 10 11 2 3 4 5 6 7 8 8 9 10 11 12 13 14

79

12 13 14 15 16 17 18 9 10 11 12 13 14 15 15 16 17 18 19 20 21

80

19 20 21 22 23 24 25 16 17 18 19 20 21 22 22 23 24 25 26 27 28

81

26 27 28 29 30 31 23 24 25 26 27 28 29 29 30 31

82

83

April May June

84

Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su

85

1 2 3 4 1 2 1 2 3 4 5 6

86

5 6 7 8 9 10 11 3 4 5 6 7 8 9 7 8 9 10 11 12 13

87

12 13 14 15 16 17 18 10 11 12 13 14 15 16 14 15 16 17 18 19 20

88

19 20 21 22 23 24 25 17 18 19 20 21 22 23 21 22 23 24 25 26 27

89

26 27 28 29 30 24 25 26 27 28 29 30 28 29 30

90

31

91

92

July August September

93

Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su

94

1 2 3 4 1 1 2 3 4 5

95

5 6 7 8 9 10 11 2 3 4 5 6 7 8 6 7 8 9 10 11 12

96

12 13 14 15 16 17 18 9 10 11 12 13 14 15 13 14 15 16 17 18 19

97

19 20 21 22 23 24 25 16 17 18 19 20 21 22 20 21 22 23 24 25 26

98

26 27 28 29 30 31 23 24 25 26 27 28 29 27 28 29 30

99

30 31

100

101

October November December

102

Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su Mo Tu We Th Fr Sa Su

103

1 2 3 1 2 3 4 5 6 7 1 2 3 4 5

104

4 5 6 7 8 9 10 8 9 10 11 12 13 14 6 7 8 9 10 11 12

105

11 12 13 14 15 16 17 15 16 17 18 19 20 21 13 14 15 16 17 18 19

106

18 19 20 21 22 23 24 22 23 24 25 26 27 28 20 21 22 23 24 25 26

107

25 26 27 28 29 30 31 29 30 27 28 29 30 31

108

"""

109

Oz N Tiram

2017-06-06 11:35:59 +0200

[diff] [blame]

110

111

default_format = dict(year="year", month="month", encoding="ascii")

112

Serhiy Storchaka

2015-02-05 15:14:35 +0200

[diff] [blame]

113

result_2004_html = """\

Oz N Tiram

2017-06-06 11:35:59 +0200

[diff] [blame]

114

<?xml version="1.0" encoding="{encoding}"?>

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

115

<!DOCTYPE html PUBLIC "-//W3C//DTD XHTML 1.0 Strict//EN" "http://www.w3.org/TR/xhtml1/DTD/xhtml1-strict.dtd">

116

<html>

117

<head>

Oz N Tiram

2017-06-06 11:35:59 +0200

[diff] [blame]

118

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

119

Thomas Wouters

47b49bf

2007-08-30 22:15:33 +0000

[diff] [blame]

120

<title>Calendar for 2004</title>

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

121

</head>

122

<body>

Oz N Tiram

2017-06-06 11:35:59 +0200

[diff] [blame]

123

124

125

<tr><th colspan="7" class="{month}">January</th></tr>

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

</table>

Oz N Tiram

2017-06-06 11:35:59 +0200

[diff] [blame]

133

</td><td><table border="0" cellpadding="0" cellspacing="0" class="{month}">

134

<tr><th colspan="7" class="{month}">February</th></tr>

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

</table>

Oz N Tiram

2017-06-06 11:35:59 +0200

[diff] [blame]

142

</td><td><table border="0" cellpadding="0" cellspacing="0" class="{month}">

143

<tr><th colspan="7" class="{month}">March</th></tr>

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

</table>

Oz N Tiram

2017-06-06 11:35:59 +0200

[diff] [blame]

151

</td></tr><tr><td><table border="0" cellpadding="0" cellspacing="0" class="{month}">

152

<tr><th colspan="7" class="{month}">April</th></tr>

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

</table>

Oz N Tiram

2017-06-06 11:35:59 +0200

[diff] [blame]

160

</td><td><table border="0" cellpadding="0" cellspacing="0" class="{month}">

161

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

</table>

Oz N Tiram

2017-06-06 11:35:59 +0200

[diff] [blame]

170

</td><td><table border="0" cellpadding="0" cellspacing="0" class="{month}">

171

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

</table>

Oz N Tiram

2017-06-06 11:35:59 +0200

[diff] [blame]

179

</td></tr><tr><td><table border="0" cellpadding="0" cellspacing="0" class="{month}">

180

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

</table>

Oz N Tiram

2017-06-06 11:35:59 +0200

[diff] [blame]

188

</td><td><table border="0" cellpadding="0" cellspacing="0" class="{month}">

189

<tr><th colspan="7" class="{month}">August</th></tr>

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

</table>

Oz N Tiram

2017-06-06 11:35:59 +0200

[diff] [blame]

198

</td><td><table border="0" cellpadding="0" cellspacing="0" class="{month}">

199

<tr><th colspan="7" class="{month}">September</th></tr>

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

</table>

Oz N Tiram

2017-06-06 11:35:59 +0200

[diff] [blame]

207

</td></tr><tr><td><table border="0" cellpadding="0" cellspacing="0" class="{month}">

208

<tr><th colspan="7" class="{month}">October</th></tr>

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

</table>

Oz N Tiram

2017-06-06 11:35:59 +0200

[diff] [blame]

216

</td><td><table border="0" cellpadding="0" cellspacing="0" class="{month}">

217

<tr><th colspan="7" class="{month}">November</th></tr>

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

</table>

Oz N Tiram

2017-06-06 11:35:59 +0200

[diff] [blame]

225

</td><td><table border="0" cellpadding="0" cellspacing="0" class="{month}">

226

<tr><th colspan="7" class="{month}">December</th></tr>

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

</table>

</td></tr></table></body>

</html>

"""

R David Murray

2012-05-29 12:55:05 -0400

[diff] [blame]

238

result_2004_days = [

239

[[[0, 0, 0, 1, 2, 3, 4],

240

[5, 6, 7, 8, 9, 10, 11],

241

[12, 13, 14, 15, 16, 17, 18],

242

[19, 20, 21, 22, 23, 24, 25],

243

[26, 27, 28, 29, 30, 31, 0]],

244

[[0, 0, 0, 0, 0, 0, 1],

245

[2, 3, 4, 5, 6, 7, 8],

246

[9, 10, 11, 12, 13, 14, 15],

247

[16, 17, 18, 19, 20, 21, 22],

248

[23, 24, 25, 26, 27, 28, 29]],

249

[[1, 2, 3, 4, 5, 6, 7],

250

[8, 9, 10, 11, 12, 13, 14],

251

[15, 16, 17, 18, 19, 20, 21],

252

[22, 23, 24, 25, 26, 27, 28],

253

[29, 30, 31, 0, 0, 0, 0]]],

254

[[[0, 0, 0, 1, 2, 3, 4],

255

[5, 6, 7, 8, 9, 10, 11],

256

[12, 13, 14, 15, 16, 17, 18],

257

[19, 20, 21, 22, 23, 24, 25],

258

[26, 27, 28, 29, 30, 0, 0]],

259

[[0, 0, 0, 0, 0, 1, 2],

260

[3, 4, 5, 6, 7, 8, 9],

261

[10, 11, 12, 13, 14, 15, 16],

262

[17, 18, 19, 20, 21, 22, 23],

263

[24, 25, 26, 27, 28, 29, 30],

264

[31, 0, 0, 0, 0, 0, 0]],

265

[[0, 1, 2, 3, 4, 5, 6],

266

[7, 8, 9, 10, 11, 12, 13],

267

[14, 15, 16, 17, 18, 19, 20],

268

[21, 22, 23, 24, 25, 26, 27],

269

[28, 29, 30, 0, 0, 0, 0]]],

270

[[[0, 0, 0, 1, 2, 3, 4],

271

[5, 6, 7, 8, 9, 10, 11],

272

[12, 13, 14, 15, 16, 17, 18],

273

[19, 20, 21, 22, 23, 24, 25],

274

[26, 27, 28, 29, 30, 31, 0]],

275

[[0, 0, 0, 0, 0, 0, 1],

276

[2, 3, 4, 5, 6, 7, 8],

277

[9, 10, 11, 12, 13, 14, 15],

278

[16, 17, 18, 19, 20, 21, 22],

279

[23, 24, 25, 26, 27, 28, 29],

280

[30, 31, 0, 0, 0, 0, 0]],

281

[[0, 0, 1, 2, 3, 4, 5],

282

[6, 7, 8, 9, 10, 11, 12],

283

[13, 14, 15, 16, 17, 18, 19],

284

[20, 21, 22, 23, 24, 25, 26],

285

[27, 28, 29, 30, 0, 0, 0]]],

286

[[[0, 0, 0, 0, 1, 2, 3],

287

[4, 5, 6, 7, 8, 9, 10],

288

[11, 12, 13, 14, 15, 16, 17],

289

[18, 19, 20, 21, 22, 23, 24],

290

[25, 26, 27, 28, 29, 30, 31]],

291

[[1, 2, 3, 4, 5, 6, 7],

292

[8, 9, 10, 11, 12, 13, 14],

293

[15, 16, 17, 18, 19, 20, 21],

294

[22, 23, 24, 25, 26, 27, 28],

295

[29, 30, 0, 0, 0, 0, 0]],

296

[[0, 0, 1, 2, 3, 4, 5],

297

[6, 7, 8, 9, 10, 11, 12],

298

[13, 14, 15, 16, 17, 18, 19],

299

[20, 21, 22, 23, 24, 25, 26],

300

[27, 28, 29, 30, 31, 0, 0]]]

301

]

302

303

result_2004_dates = \

304

[[['12/29/03 12/30/03 12/31/03 01/01/04 01/02/04 01/03/04 01/04/04',

305

'01/05/04 01/06/04 01/07/04 01/08/04 01/09/04 01/10/04 01/11/04',

306

'01/12/04 01/13/04 01/14/04 01/15/04 01/16/04 01/17/04 01/18/04',

307

'01/19/04 01/20/04 01/21/04 01/22/04 01/23/04 01/24/04 01/25/04',

308

'01/26/04 01/27/04 01/28/04 01/29/04 01/30/04 01/31/04 02/01/04'],

309

['01/26/04 01/27/04 01/28/04 01/29/04 01/30/04 01/31/04 02/01/04',

310

'02/02/04 02/03/04 02/04/04 02/05/04 02/06/04 02/07/04 02/08/04',

311

'02/09/04 02/10/04 02/11/04 02/12/04 02/13/04 02/14/04 02/15/04',

312

'02/16/04 02/17/04 02/18/04 02/19/04 02/20/04 02/21/04 02/22/04',

313

'02/23/04 02/24/04 02/25/04 02/26/04 02/27/04 02/28/04 02/29/04'],

314

['03/01/04 03/02/04 03/03/04 03/04/04 03/05/04 03/06/04 03/07/04',

315

'03/08/04 03/09/04 03/10/04 03/11/04 03/12/04 03/13/04 03/14/04',

316

'03/15/04 03/16/04 03/17/04 03/18/04 03/19/04 03/20/04 03/21/04',

317

'03/22/04 03/23/04 03/24/04 03/25/04 03/26/04 03/27/04 03/28/04',

318

'03/29/04 03/30/04 03/31/04 04/01/04 04/02/04 04/03/04 04/04/04']],

319

[['03/29/04 03/30/04 03/31/04 04/01/04 04/02/04 04/03/04 04/04/04',

320

'04/05/04 04/06/04 04/07/04 04/08/04 04/09/04 04/10/04 04/11/04',

321

'04/12/04 04/13/04 04/14/04 04/15/04 04/16/04 04/17/04 04/18/04',

322

'04/19/04 04/20/04 04/21/04 04/22/04 04/23/04 04/24/04 04/25/04',

323

'04/26/04 04/27/04 04/28/04 04/29/04 04/30/04 05/01/04 05/02/04'],

324

['04/26/04 04/27/04 04/28/04 04/29/04 04/30/04 05/01/04 05/02/04',

325

'05/03/04 05/04/04 05/05/04 05/06/04 05/07/04 05/08/04 05/09/04',

326

'05/10/04 05/11/04 05/12/04 05/13/04 05/14/04 05/15/04 05/16/04',

327

'05/17/04 05/18/04 05/19/04 05/20/04 05/21/04 05/22/04 05/23/04',

328

'05/24/04 05/25/04 05/26/04 05/27/04 05/28/04 05/29/04 05/30/04',

329

'05/31/04 06/01/04 06/02/04 06/03/04 06/04/04 06/05/04 06/06/04'],

330

['05/31/04 06/01/04 06/02/04 06/03/04 06/04/04 06/05/04 06/06/04',

331

'06/07/04 06/08/04 06/09/04 06/10/04 06/11/04 06/12/04 06/13/04',

332

'06/14/04 06/15/04 06/16/04 06/17/04 06/18/04 06/19/04 06/20/04',

333

'06/21/04 06/22/04 06/23/04 06/24/04 06/25/04 06/26/04 06/27/04',

334

'06/28/04 06/29/04 06/30/04 07/01/04 07/02/04 07/03/04 07/04/04']],

335

[['06/28/04 06/29/04 06/30/04 07/01/04 07/02/04 07/03/04 07/04/04',

336

'07/05/04 07/06/04 07/07/04 07/08/04 07/09/04 07/10/04 07/11/04',

337

'07/12/04 07/13/04 07/14/04 07/15/04 07/16/04 07/17/04 07/18/04',

338

'07/19/04 07/20/04 07/21/04 07/22/04 07/23/04 07/24/04 07/25/04',

339

'07/26/04 07/27/04 07/28/04 07/29/04 07/30/04 07/31/04 08/01/04'],

340

['07/26/04 07/27/04 07/28/04 07/29/04 07/30/04 07/31/04 08/01/04',

341

'08/02/04 08/03/04 08/04/04 08/05/04 08/06/04 08/07/04 08/08/04',

342

'08/09/04 08/10/04 08/11/04 08/12/04 08/13/04 08/14/04 08/15/04',

343

'08/16/04 08/17/04 08/18/04 08/19/04 08/20/04 08/21/04 08/22/04',

344

'08/23/04 08/24/04 08/25/04 08/26/04 08/27/04 08/28/04 08/29/04',

345

'08/30/04 08/31/04 09/01/04 09/02/04 09/03/04 09/04/04 09/05/04'],

346

['08/30/04 08/31/04 09/01/04 09/02/04 09/03/04 09/04/04 09/05/04',

347

'09/06/04 09/07/04 09/08/04 09/09/04 09/10/04 09/11/04 09/12/04',

348

'09/13/04 09/14/04 09/15/04 09/16/04 09/17/04 09/18/04 09/19/04',

349

'09/20/04 09/21/04 09/22/04 09/23/04 09/24/04 09/25/04 09/26/04',

350

'09/27/04 09/28/04 09/29/04 09/30/04 10/01/04 10/02/04 10/03/04']],

351

[['09/27/04 09/28/04 09/29/04 09/30/04 10/01/04 10/02/04 10/03/04',

352

'10/04/04 10/05/04 10/06/04 10/07/04 10/08/04 10/09/04 10/10/04',

353

'10/11/04 10/12/04 10/13/04 10/14/04 10/15/04 10/16/04 10/17/04',

354

'10/18/04 10/19/04 10/20/04 10/21/04 10/22/04 10/23/04 10/24/04',

355

'10/25/04 10/26/04 10/27/04 10/28/04 10/29/04 10/30/04 10/31/04'],

356

['11/01/04 11/02/04 11/03/04 11/04/04 11/05/04 11/06/04 11/07/04',

357

'11/08/04 11/09/04 11/10/04 11/11/04 11/12/04 11/13/04 11/14/04',

358

'11/15/04 11/16/04 11/17/04 11/18/04 11/19/04 11/20/04 11/21/04',

359

'11/22/04 11/23/04 11/24/04 11/25/04 11/26/04 11/27/04 11/28/04',

360

'11/29/04 11/30/04 12/01/04 12/02/04 12/03/04 12/04/04 12/05/04'],

361

['11/29/04 11/30/04 12/01/04 12/02/04 12/03/04 12/04/04 12/05/04',

362

'12/06/04 12/07/04 12/08/04 12/09/04 12/10/04 12/11/04 12/12/04',

363

'12/13/04 12/14/04 12/15/04 12/16/04 12/17/04 12/18/04 12/19/04',

364

'12/20/04 12/21/04 12/22/04 12/23/04 12/24/04 12/25/04 12/26/04',

365

'12/27/04 12/28/04 12/29/04 12/30/04 12/31/04 01/01/05 01/02/05']]]

366

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

367

368

class OutputTestCase(unittest.TestCase):

369

def normalize_calendar(self, s):

Mark Dickinson

934896d

2009-02-21 20:59:32 +0000

[diff] [blame]

370

# Filters out locale dependent strings

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

371

def neitherspacenordigit(c):

372

return not c.isspace() and not c.isdigit()

373

374

lines = []

Ezio Melotti

d8b509b

2011-09-28 17:37:55 +0300

[diff] [blame]

375

for line in s.splitlines(keepends=False):

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

376

# Drop texts, as they are locale dependent

377

if line and not filter(neitherspacenordigit, line):

lines.append(line)

return lines

R David Murray

2012-05-29 12:55:05 -0400

[diff] [blame]

381

def check_htmlcalendar_encoding(self, req, res):

382

cal = calendar.HTMLCalendar()

Oz N Tiram

2017-06-06 11:35:59 +0200

[diff] [blame]

383

format_ = default_format.copy()

384

format_["encoding"] = req or 'utf-8'

385

output = cal.formatyearpage(2004, encoding=req)

R David Murray

2012-05-29 12:55:05 -0400

[diff] [blame]

386

self.assertEqual(

Oz N Tiram

2017-06-06 11:35:59 +0200

[diff] [blame]

387

output,

388

result_2004_html.format(**format_).encode(res)

R David Murray

2012-05-29 12:55:05 -0400

[diff] [blame]

389

)

390

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

391

def test_output(self):

392

self.assertEqual(

393

self.normalize_calendar(calendar.calendar(2004)),

394

self.normalize_calendar(result_2004_text)

395

)

Alexander Belopolsky

2017-10-26 15:34:11 -0400

[diff] [blame]

396

self.assertEqual(

397

self.normalize_calendar(calendar.calendar(0)),

398

self.normalize_calendar(result_0_text)

399

)

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

400

401

def test_output_textcalendar(self):

402

self.assertEqual(

Serhiy Storchaka

2015-02-05 15:14:35 +0200

[diff] [blame]

403

calendar.TextCalendar().formatyear(2004),

404

result_2004_text

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

405

)

Alexander Belopolsky

2017-10-26 15:34:11 -0400

[diff] [blame]

406

self.assertEqual(

407

calendar.TextCalendar().formatyear(0),

408

result_0_text

409

)

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

410

R David Murray

2012-05-29 12:55:05 -0400

[diff] [blame]

411

def test_output_htmlcalendar_encoding_ascii(self):

412

self.check_htmlcalendar_encoding('ascii', 'ascii')

413

414

def test_output_htmlcalendar_encoding_utf8(self):

415

self.check_htmlcalendar_encoding('utf-8', 'utf-8')

416

417

def test_output_htmlcalendar_encoding_default(self):

418

self.check_htmlcalendar_encoding(None, sys.getdefaultencoding())

419

420

def test_yeardatescalendar(self):

421

def shrink(cal):

R David Murray

bb75678

2012-05-30 08:10:54 -0400

[diff] [blame]

422

return [[[' '.join('{:02d}/{:02d}/{}'.format(

423

d.month, d.day, str(d.year)[-2:]) for d in z)

424

for z in y] for y in x] for x in cal]

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

425

self.assertEqual(

R David Murray

2012-05-29 12:55:05 -0400

[diff] [blame]

426

shrink(calendar.Calendar().yeardatescalendar(2004)),

result_2004_dates

)

def test_yeardayscalendar(self):

431

self.assertEqual(

432

calendar.Calendar().yeardayscalendar(2004),

result_2004_days

)

def test_formatweekheader_short(self):

437

self.assertEqual(

438

calendar.TextCalendar().formatweekheader(2),

439

'Mo Tu We Th Fr Sa Su'

440

)

441

442

def test_formatweekheader_long(self):

443

self.assertEqual(

444

calendar.TextCalendar().formatweekheader(9),

445

' Monday Tuesday Wednesday Thursday '

446

' Friday Saturday Sunday '

447

)

448

449

def test_formatmonth(self):

450

self.assertEqual(

Serhiy Storchaka

2015-02-05 15:14:35 +0200

[diff] [blame]

451

calendar.TextCalendar().formatmonth(2004, 1),

452

result_2004_01_text

R David Murray

2012-05-29 12:55:05 -0400

[diff] [blame]

453

)

Alexander Belopolsky

2017-10-26 15:34:11 -0400

[diff] [blame]

454

self.assertEqual(

455

calendar.TextCalendar().formatmonth(0, 2),

456

result_0_02_text

457

)

R David Murray

2012-05-29 12:55:05 -0400

[diff] [blame]

458

459

def test_formatmonthname_with_year(self):

460

self.assertEqual(

461

calendar.HTMLCalendar().formatmonthname(2004, 1, withyear=True),

462

'<tr><th colspan="7" class="month">January 2004</th></tr>'

463

)

464

465

def test_formatmonthname_without_year(self):

466

self.assertEqual(

467

calendar.HTMLCalendar().formatmonthname(2004, 1, withyear=False),

468

'<tr><th colspan="7" class="month">January</th></tr>'

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

469

)

470

Ezio Melotti

2013-09-14 04:55:53 +0300

[diff] [blame]

471

def test_prweek(self):

472

with support.captured_stdout() as out:

473

week = [(1,0), (2,1), (3,2), (4,3), (5,4), (6,5), (7,6)]

474

calendar.TextCalendar().prweek(week, 1)

Serhiy Storchaka

0595ed2

2016-10-25 15:20:58 +0300

[diff] [blame]

475

self.assertEqual(out.getvalue(), " 1 2 3 4 5 6 7")

Ezio Melotti

2013-09-14 04:55:53 +0300

[diff] [blame]

476

477

def test_prmonth(self):

478

with support.captured_stdout() as out:

479

calendar.TextCalendar().prmonth(2004, 1)

Serhiy Storchaka

7ff51bd

2016-10-25 15:00:52 +0300

[diff] [blame]

480

self.assertEqual(out.getvalue(), result_2004_01_text)

Ezio Melotti

2013-09-14 04:55:53 +0300

[diff] [blame]

481

482

def test_pryear(self):

483

with support.captured_stdout() as out:

484

calendar.TextCalendar().pryear(2004)

Serhiy Storchaka

0595ed2

2016-10-25 15:20:58 +0300

[diff] [blame]

485

self.assertEqual(out.getvalue(), result_2004_text)

Ezio Melotti

2013-09-14 04:55:53 +0300

[diff] [blame]

486

487

def test_format(self):

488

with support.captured_stdout() as out:

489

calendar.format(["1", "2", "3"], colwidth=3, spacing=1)

490

self.assertEqual(out.getvalue().strip(), "1 2 3")

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

491

Fred Drake

2001-12-12 05:38:08 +0000

[diff] [blame]

492

class CalendarTestCase(unittest.TestCase):

493

def test_isleap(self):

494

# Make sure that the return is right for a few years, and

495

# ensure that the return values are 1 or 0, not just true or

496

# false (see SF bug #485794). Specific additional tests may

497

# be appropriate; this tests a single "cycle".

498

self.assertEqual(calendar.isleap(2000), 1)

499

self.assertEqual(calendar.isleap(2001), 0)

500

self.assertEqual(calendar.isleap(2002), 0)

501

self.assertEqual(calendar.isleap(2003), 0)

502

503

def test_setfirstweekday(self):

Guido van Rossum

47b9ff6

2006-08-24 00:41:19 +0000

[diff] [blame]

504

self.assertRaises(TypeError, calendar.setfirstweekday, 'flabber')

Fred Drake

2001-12-12 05:38:08 +0000

[diff] [blame]

505

self.assertRaises(ValueError, calendar.setfirstweekday, -1)

506

self.assertRaises(ValueError, calendar.setfirstweekday, 200)

507

orig = calendar.firstweekday()

508

calendar.setfirstweekday(calendar.SUNDAY)

509

self.assertEqual(calendar.firstweekday(), calendar.SUNDAY)

510

calendar.setfirstweekday(calendar.MONDAY)

511

self.assertEqual(calendar.firstweekday(), calendar.MONDAY)

512

calendar.setfirstweekday(orig)

513

R David Murray

2012-05-29 12:55:05 -0400

[diff] [blame]

514

def test_illegal_weekday_reported(self):

515

with self.assertRaisesRegex(calendar.IllegalWeekdayError, '123'):

516

calendar.setfirstweekday(123)

517

518

def test_enumerate_weekdays(self):

Skip Montanaro

693c6c4

2002-03-15 03:57:04 +0000

[diff] [blame]

519

self.assertRaises(IndexError, calendar.day_abbr.__getitem__, -10)

520

self.assertRaises(IndexError, calendar.day_name.__getitem__, 10)

521

self.assertEqual(len([d for d in calendar.day_abbr]), 7)

Fred Drake

2001-12-12 05:38:08 +0000

[diff] [blame]

522

Tim Peters

0c2c8e7

2002-03-23 03:26:53 +0000

[diff] [blame]

523

def test_days(self):

524

for attr in "day_name", "day_abbr":

525

value = getattr(calendar, attr)

526

self.assertEqual(len(value), 7)

527

self.assertEqual(len(value[:]), 7)

528

# ensure they're all unique

Tim Peters

bbc0d44

2004-11-13 16:18:32 +0000

[diff] [blame]

529

self.assertEqual(len(set(value)), 7)

530

# verify it "acts like a sequence" in two forms of iteration

531

self.assertEqual(value[::-1], list(reversed(value)))

Tim Peters

0c2c8e7

2002-03-23 03:26:53 +0000

[diff] [blame]

532

533

def test_months(self):

534

for attr in "month_name", "month_abbr":

535

value = getattr(calendar, attr)

536

self.assertEqual(len(value), 13)

537

self.assertEqual(len(value[:]), 13)

538

self.assertEqual(value[0], "")

539

# ensure they're all unique

Tim Peters

bbc0d44

2004-11-13 16:18:32 +0000

[diff] [blame]

540

self.assertEqual(len(set(value)), 13)

541

# verify it "acts like a sequence" in two forms of iteration

542

self.assertEqual(value[::-1], list(reversed(value)))

Tim Peters

0c2c8e7

2002-03-23 03:26:53 +0000

[diff] [blame]

543

R David Murray

2012-05-29 12:55:05 -0400

[diff] [blame]

544

def test_locale_calendars(self):

Georg Brandl

2010-10-19 18:54:25 +0000

[diff] [blame]

545

# ensure that Locale{Text,HTML}Calendar resets the locale properly

546

# (it is still not thread-safe though)

Georg Brandl

2010-10-19 18:54:25 +0000

[diff] [blame]

547

old_october = calendar.TextCalendar().formatmonthname(2010, 10, 10)

Georg Brandl

9450cc0

2010-10-20 06:50:19 +0000

[diff] [blame]

548

try:

Serhiy Storchaka

f9fcdb7

2013-01-31 16:00:42 +0200

[diff] [blame]

549

cal = calendar.LocaleTextCalendar(locale='')

550

local_weekday = cal.formatweekday(1, 10)

551

local_month = cal.formatmonthname(2010, 10, 10)

Georg Brandl

9450cc0

2010-10-20 06:50:19 +0000

[diff] [blame]

552

except locale.Error:

553

# cannot set the system default locale -- skip rest of test

Serhiy Storchaka

f9fcdb7

2013-01-31 16:00:42 +0200

[diff] [blame]

554

raise unittest.SkipTest('cannot set the system default locale')

555

self.assertIsInstance(local_weekday, str)

556

self.assertIsInstance(local_month, str)

557

self.assertEqual(len(local_weekday), 10)

558

self.assertGreaterEqual(len(local_month), 10)

559

cal = calendar.LocaleHTMLCalendar(locale='')

560

local_weekday = cal.formatweekday(1)

561

local_month = cal.formatmonthname(2010, 10)

562

self.assertIsInstance(local_weekday, str)

563

self.assertIsInstance(local_month, str)

Georg Brandl

2010-10-19 18:54:25 +0000

[diff] [blame]

564

new_october = calendar.TextCalendar().formatmonthname(2010, 10, 10)

Ezio Melotti

b3aedd4

2010-11-20 19:04:17 +0000

[diff] [blame]

565

self.assertEqual(old_october, new_october)

Georg Brandl

2010-10-19 18:54:25 +0000

[diff] [blame]

566

Srinivas Reddy Thatiparthy (శ్రీనివాస్ రెడ్డి తాటిపర్తి)

85339f5

2020-06-02 17:03:09 +0530

[diff] [blame]

567

def test_locale_html_calendar_custom_css_class_month_name(self):

568

try:

569

cal = calendar.LocaleHTMLCalendar(locale='')

570

local_month = cal.formatmonthname(2010, 10, 10)

571

except locale.Error:

572

# cannot set the system default locale -- skip rest of test

573

raise unittest.SkipTest('cannot set the system default locale')

574

self.assertIn('class="month"', local_month)

575

cal.cssclass_month_head = "text-center month"

576

local_month = cal.formatmonthname(2010, 10, 10)

577

self.assertIn('class="text-center month"', local_month)

578

579

def test_locale_html_calendar_custom_css_class_weekday(self):

580

try:

581

cal = calendar.LocaleHTMLCalendar(locale='')

582

local_weekday = cal.formatweekday(6)

583

except locale.Error:

584

# cannot set the system default locale -- skip rest of test

585

raise unittest.SkipTest('cannot set the system default locale')

586

self.assertIn('class="sun"', local_weekday)

587

cal.cssclasses_weekday_head = ["mon2", "tue2", "wed2", "thu2", "fri2", "sat2", "sun2"]

588

local_weekday = cal.formatweekday(6)

589

self.assertIn('class="sun2"', local_weekday)

590

Alexander Belopolsky

fdd9b21

2017-10-24 13:17:10 -0400

[diff] [blame]

591

def test_itermonthdays3(self):

592

# ensure itermonthdays3 doesn't overflow after datetime.MAXYEAR

593

list(calendar.Calendar().itermonthdays3(datetime.MAXYEAR, 12))

594

595

def test_itermonthdays4(self):

596

cal = calendar.Calendar(firstweekday=3)

597

days = list(cal.itermonthdays4(2001, 2))

598

self.assertEqual(days[0], (2001, 2, 1, 3))

599

self.assertEqual(days[-1], (2001, 2, 28, 2))

Ezio Melotti

85710a4

2012-09-21 17:26:35 +0300

[diff] [blame]

600

Alexander Belopolsky

957b756

2016-09-27 20:26:39 -0400

[diff] [blame]

601

def test_itermonthdays(self):

602

for firstweekday in range(7):

603

cal = calendar.Calendar(firstweekday)

604

# Test the extremes, see #28253 and #26650

605

for y, m in [(1, 1), (9999, 12)]:

606

days = list(cal.itermonthdays(y, m))

607

self.assertIn(len(days), (35, 42))

608

# Test a short month

609

cal = calendar.Calendar(firstweekday=3)

610

days = list(cal.itermonthdays(2001, 2))

611

self.assertEqual(days, list(range(1, 29)))

612

613

def test_itermonthdays2(self):

614

for firstweekday in range(7):

615

cal = calendar.Calendar(firstweekday)

616

# Test the extremes, see #28253 and #26650

617

for y, m in [(1, 1), (9999, 12)]:

618

days = list(cal.itermonthdays2(y, m))

619

self.assertEqual(days[0][1], firstweekday)

620

self.assertEqual(days[-1][1], (firstweekday - 1) % 7)

621

Walter Dörwald

2004-07-02 19:00:09 +0000

[diff] [blame]

622

623

class MonthCalendarTestCase(unittest.TestCase):

624

def setUp(self):

625

self.oldfirstweekday = calendar.firstweekday()

626

calendar.setfirstweekday(self.firstweekday)

627

628

def tearDown(self):

629

calendar.setfirstweekday(self.oldfirstweekday)

630

631

def check_weeks(self, year, month, weeks):

632

cal = calendar.monthcalendar(year, month)

633

self.assertEqual(len(cal), len(weeks))

Guido van Rossum

805365e

2007-05-07 22:24:25 +0000

[diff] [blame]

634

for i in range(len(weeks)):

Walter Dörwald

2004-07-02 19:00:09 +0000

[diff] [blame]

635

self.assertEqual(weeks[i], sum(day != 0 for day in cal[i]))

636

637

638

class MondayTestCase(MonthCalendarTestCase):

639

firstweekday = calendar.MONDAY

640

641

def test_february(self):

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

642

# A 28-day february starting on monday (7+7+7+7 days)

Walter Dörwald

2004-07-02 19:00:09 +0000

[diff] [blame]

643

self.check_weeks(1999, 2, (7, 7, 7, 7))

644

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

645

# A 28-day february starting on tuesday (6+7+7+7+1 days)

Walter Dörwald

2004-07-02 19:00:09 +0000

[diff] [blame]

646

self.check_weeks(2005, 2, (6, 7, 7, 7, 1))

647

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

648

# A 28-day february starting on sunday (1+7+7+7+6 days)

Walter Dörwald

2004-07-02 19:00:09 +0000

[diff] [blame]

649

self.check_weeks(1987, 2, (1, 7, 7, 7, 6))

650

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

651

# A 29-day february starting on monday (7+7+7+7+1 days)

Walter Dörwald

2004-07-02 19:00:09 +0000

[diff] [blame]

652

self.check_weeks(1988, 2, (7, 7, 7, 7, 1))

653

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

654

# A 29-day february starting on tuesday (6+7+7+7+2 days)

Walter Dörwald

2004-07-02 19:00:09 +0000

[diff] [blame]

655

self.check_weeks(1972, 2, (6, 7, 7, 7, 2))

656

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

657

# A 29-day february starting on sunday (1+7+7+7+7 days)

Walter Dörwald

2004-07-02 19:00:09 +0000

[diff] [blame]

658

self.check_weeks(2004, 2, (1, 7, 7, 7, 7))

659

660

def test_april(self):

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

661

# A 30-day april starting on monday (7+7+7+7+2 days)

Walter Dörwald

2004-07-02 19:00:09 +0000

[diff] [blame]

662

self.check_weeks(1935, 4, (7, 7, 7, 7, 2))

663

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

664

# A 30-day april starting on tuesday (6+7+7+7+3 days)

Walter Dörwald

2004-07-02 19:00:09 +0000

[diff] [blame]

665

self.check_weeks(1975, 4, (6, 7, 7, 7, 3))

666

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

667

# A 30-day april starting on sunday (1+7+7+7+7+1 days)

Walter Dörwald

2004-07-02 19:00:09 +0000

[diff] [blame]

668

self.check_weeks(1945, 4, (1, 7, 7, 7, 7, 1))

669

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

670

# A 30-day april starting on saturday (2+7+7+7+7 days)

Walter Dörwald

2004-07-02 19:00:09 +0000

[diff] [blame]

671

self.check_weeks(1995, 4, (2, 7, 7, 7, 7))

672

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

673

# A 30-day april starting on friday (3+7+7+7+6 days)

Walter Dörwald

2004-07-02 19:00:09 +0000

[diff] [blame]

674

self.check_weeks(1994, 4, (3, 7, 7, 7, 6))

675

676

def test_december(self):

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

677

# A 31-day december starting on monday (7+7+7+7+3 days)

Walter Dörwald

2004-07-02 19:00:09 +0000

[diff] [blame]

678

self.check_weeks(1980, 12, (7, 7, 7, 7, 3))

679

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

680

# A 31-day december starting on tuesday (6+7+7+7+4 days)

Walter Dörwald

2004-07-02 19:00:09 +0000

[diff] [blame]

681

self.check_weeks(1987, 12, (6, 7, 7, 7, 4))

682

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

683

# A 31-day december starting on sunday (1+7+7+7+7+2 days)

Walter Dörwald

2004-07-02 19:00:09 +0000

[diff] [blame]

684

self.check_weeks(1968, 12, (1, 7, 7, 7, 7, 2))

685

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

686

# A 31-day december starting on thursday (4+7+7+7+6 days)

Walter Dörwald

2004-07-02 19:00:09 +0000

[diff] [blame]

687

self.check_weeks(1988, 12, (4, 7, 7, 7, 6))

688

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

689

# A 31-day december starting on friday (3+7+7+7+7 days)

Walter Dörwald

2004-07-02 19:00:09 +0000

[diff] [blame]

690

self.check_weeks(2017, 12, (3, 7, 7, 7, 7))

691

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

692

# A 31-day december starting on saturday (2+7+7+7+7+1 days)

Walter Dörwald

2004-07-02 19:00:09 +0000

[diff] [blame]

693

self.check_weeks(2068, 12, (2, 7, 7, 7, 7, 1))

694

695

Walter Dörwald

2004-07-08 17:14:17 +0000

[diff] [blame]

696

class SundayTestCase(MonthCalendarTestCase):

697

firstweekday = calendar.SUNDAY

698

699

def test_february(self):

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

700

# A 28-day february starting on sunday (7+7+7+7 days)

Walter Dörwald

2004-07-08 17:14:17 +0000

[diff] [blame]

701

self.check_weeks(2009, 2, (7, 7, 7, 7))

702

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

703

# A 28-day february starting on monday (6+7+7+7+1 days)

Walter Dörwald

2004-07-08 17:14:17 +0000

[diff] [blame]

704

self.check_weeks(1999, 2, (6, 7, 7, 7, 1))

705

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

706

# A 28-day february starting on saturday (1+7+7+7+6 days)

Walter Dörwald

2004-07-08 17:14:17 +0000

[diff] [blame]

707

self.check_weeks(1997, 2, (1, 7, 7, 7, 6))

708

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

709

# A 29-day february starting on sunday (7+7+7+7+1 days)

Walter Dörwald

2004-07-08 17:14:17 +0000

[diff] [blame]

710

self.check_weeks(2004, 2, (7, 7, 7, 7, 1))

711

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

712

# A 29-day february starting on monday (6+7+7+7+2 days)

Walter Dörwald

2004-07-08 17:14:17 +0000

[diff] [blame]

713

self.check_weeks(1960, 2, (6, 7, 7, 7, 2))

714

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

715

# A 29-day february starting on saturday (1+7+7+7+7 days)

Walter Dörwald

2004-07-08 17:14:17 +0000

[diff] [blame]

716

self.check_weeks(1964, 2, (1, 7, 7, 7, 7))

717

718

def test_april(self):

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

719

# A 30-day april starting on sunday (7+7+7+7+2 days)

Walter Dörwald

2004-07-08 17:14:17 +0000

[diff] [blame]

720

self.check_weeks(1923, 4, (7, 7, 7, 7, 2))

721

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

722

# A 30-day april starting on monday (6+7+7+7+3 days)

Walter Dörwald

2004-07-08 17:14:17 +0000

[diff] [blame]

723

self.check_weeks(1918, 4, (6, 7, 7, 7, 3))

724

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

725

# A 30-day april starting on saturday (1+7+7+7+7+1 days)

Walter Dörwald

2004-07-08 17:14:17 +0000

[diff] [blame]

726

self.check_weeks(1950, 4, (1, 7, 7, 7, 7, 1))

727

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

728

# A 30-day april starting on friday (2+7+7+7+7 days)

Walter Dörwald

2004-07-08 17:14:17 +0000

[diff] [blame]

729

self.check_weeks(1960, 4, (2, 7, 7, 7, 7))

730

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

731

# A 30-day april starting on thursday (3+7+7+7+6 days)

Walter Dörwald

2004-07-08 17:14:17 +0000

[diff] [blame]

732

self.check_weeks(1909, 4, (3, 7, 7, 7, 6))

733

734

def test_december(self):

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

735

# A 31-day december starting on sunday (7+7+7+7+3 days)

Walter Dörwald

2004-07-08 17:14:17 +0000

[diff] [blame]

736

self.check_weeks(2080, 12, (7, 7, 7, 7, 3))

737

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

738

# A 31-day december starting on monday (6+7+7+7+4 days)

Walter Dörwald

2004-07-08 17:14:17 +0000

[diff] [blame]

739

self.check_weeks(1941, 12, (6, 7, 7, 7, 4))

740

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

741

# A 31-day december starting on saturday (1+7+7+7+7+2 days)

Walter Dörwald

2004-07-08 17:14:17 +0000

[diff] [blame]

742

self.check_weeks(1923, 12, (1, 7, 7, 7, 7, 2))

743

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

744

# A 31-day december starting on wednesday (4+7+7+7+6 days)

Walter Dörwald

2004-07-08 17:14:17 +0000

[diff] [blame]

745

self.check_weeks(1948, 12, (4, 7, 7, 7, 6))

746

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

747

# A 31-day december starting on thursday (3+7+7+7+7 days)

Walter Dörwald

2004-07-08 17:14:17 +0000

[diff] [blame]

748

self.check_weeks(1927, 12, (3, 7, 7, 7, 7))

749

Thomas Wouters

2006-04-21 10:40:58 +0000

[diff] [blame]

750

# A 31-day december starting on friday (2+7+7+7+7+1 days)

Walter Dörwald

2004-07-08 17:14:17 +0000

[diff] [blame]

751

self.check_weeks(1995, 12, (2, 7, 7, 7, 7, 1))

752

Alexander Belopolsky

43ca710

2010-06-14 17:32:03 +0000

[diff] [blame]

753

class TimegmTestCase(unittest.TestCase):

754

TIMESTAMPS = [0, 10, 100, 1000, 10000, 100000, 1000000,

755

1234567890, 1262304000, 1275785153,]

756

def test_timegm(self):

757

for secs in self.TIMESTAMPS:

758

tuple = time.gmtime(secs)

759

self.assertEqual(secs, calendar.timegm(tuple))

Walter Dörwald

2004-07-08 17:14:17 +0000

[diff] [blame]

760

Brett Cannon

cfae36f

2010-07-23 13:54:14 +0000

[diff] [blame]

761

class MonthRangeTestCase(unittest.TestCase):

762

def test_january(self):

763

# Tests valid lower boundary case.

764

self.assertEqual(calendar.monthrange(2004,1), (3,31))

765

766

def test_february_leap(self):

767

# Tests February during leap year.

768

self.assertEqual(calendar.monthrange(2004,2), (6,29))

769

770

def test_february_nonleap(self):

771

# Tests February in non-leap year.

772

self.assertEqual(calendar.monthrange(2010,2), (0,28))

773

774

def test_december(self):

775

# Tests valid upper boundary case.

776

self.assertEqual(calendar.monthrange(2004,12), (2,31))

777

778

def test_zeroth_month(self):

779

# Tests low invalid boundary case.

780

with self.assertRaises(calendar.IllegalMonthError):

781

calendar.monthrange(2004, 0)

782

783

def test_thirteenth_month(self):

784

# Tests high invalid boundary case.

785

with self.assertRaises(calendar.IllegalMonthError):

786

calendar.monthrange(2004, 13)

787

R David Murray

2012-05-29 12:55:05 -0400

[diff] [blame]

788

def test_illegal_month_reported(self):

789

with self.assertRaisesRegex(calendar.IllegalMonthError, '65'):

790

calendar.monthrange(2004, 65)

791

R. David Murray

1daaf9e

2010-12-31 19:21:14 +0000

[diff] [blame]

792

class LeapdaysTestCase(unittest.TestCase):

793

def test_no_range(self):

794

# test when no range i.e. two identical years as args

795

self.assertEqual(calendar.leapdays(2010,2010), 0)

796

797

def test_no_leapdays(self):

798

# test when no leap years in range

799

self.assertEqual(calendar.leapdays(2010,2011), 0)

800

801

def test_no_leapdays_upper_boundary(self):

802

# test no leap years in range, when upper boundary is a leap year

803

self.assertEqual(calendar.leapdays(2010,2012), 0)

804

805

def test_one_leapday_lower_boundary(self):

806

# test when one leap year in range, lower boundary is leap year

807

self.assertEqual(calendar.leapdays(2012,2013), 1)

808

809

def test_several_leapyears_in_range(self):

810

self.assertEqual(calendar.leapdays(1997,2020), 5)

811

Brett Cannon

cfae36f

2010-07-23 13:54:14 +0000

[diff] [blame]

812

Serhiy Storchaka

2015-02-05 15:14:35 +0200

[diff] [blame]

813

def conv(s):

814

return s.replace('\n', os.linesep).encode()

Senthil Kumaran

962fed9

2011-08-11 09:22:52 +0800

[diff] [blame]

815

Serhiy Storchaka

2015-02-05 15:14:35 +0200

[diff] [blame]

816

class CommandLineTestCase(unittest.TestCase):

817

def run_ok(self, *args):

818

return assert_python_ok('-m', 'calendar', *args)[1]

819

820

def assertFailure(self, *args):

821

rc, stdout, stderr = assert_python_failure('-m', 'calendar', *args)

Serhiy Storchaka

9785261

2015-11-01 17:14:27 +0200

[diff] [blame]

822

self.assertIn(b'usage:', stderr)

Serhiy Storchaka

2015-02-05 15:14:35 +0200

[diff] [blame]

823

self.assertEqual(rc, 2)

824

825

def test_help(self):

826

stdout = self.run_ok('-h')

Serhiy Storchaka

9785261

2015-11-01 17:14:27 +0200

[diff] [blame]

827

self.assertIn(b'usage:', stdout)

Serhiy Storchaka

2015-02-05 15:14:35 +0200

[diff] [blame]

828

self.assertIn(b'calendar.py', stdout)

829

self.assertIn(b'--help', stdout)

830

831

def test_illegal_arguments(self):

832

self.assertFailure('-z')

Serhiy Storchaka

9785261

2015-11-01 17:14:27 +0200

[diff] [blame]

833

self.assertFailure('spam')

834

self.assertFailure('2004', 'spam')

Serhiy Storchaka

2015-02-05 15:14:35 +0200

[diff] [blame]

835

self.assertFailure('-t', 'html', '2004', '1')

836

837

def test_output_current_year(self):

838

stdout = self.run_ok()

839

year = datetime.datetime.now().year

840

self.assertIn((' %s' % year).encode(), stdout)

841

self.assertIn(b'January', stdout)

842

self.assertIn(b'Mo Tu We Th Fr Sa Su', stdout)

843

844

def test_output_year(self):

845

stdout = self.run_ok('2004')

846

self.assertEqual(stdout, conv(result_2004_text))

847

848

def test_output_month(self):

849

stdout = self.run_ok('2004', '1')

850

self.assertEqual(stdout, conv(result_2004_01_text))

851

852

def test_option_encoding(self):

853

self.assertFailure('-e')

854

self.assertFailure('--encoding')

855

stdout = self.run_ok('--encoding', 'utf-16-le', '2004')

856

self.assertEqual(stdout, result_2004_text.encode('utf-16-le'))

857

858

def test_option_locale(self):

859

self.assertFailure('-L')

860

self.assertFailure('--locale')

861

self.assertFailure('-L', 'en')

862

lang, enc = locale.getdefaultlocale()

lang = lang or 'C'

enc = enc or 'UTF-8'

try:

oldlocale = locale.getlocale(locale.LC_TIME)

867

try:

868

locale.setlocale(locale.LC_TIME, (lang, enc))

869

finally:

870

locale.setlocale(locale.LC_TIME, oldlocale)

871

except (locale.Error, ValueError):

872

self.skipTest('cannot set the system default locale')

873

stdout = self.run_ok('--locale', lang, '--encoding', enc, '2004')

874

self.assertIn('2004'.encode(enc), stdout)

875

876

def test_option_width(self):

877

self.assertFailure('-w')

878

self.assertFailure('--width')

879

self.assertFailure('-w', 'spam')

880

stdout = self.run_ok('--width', '3', '2004')

881

self.assertIn(b'Mon Tue Wed Thu Fri Sat Sun', stdout)

882

883

def test_option_lines(self):

884

self.assertFailure('-l')

885

self.assertFailure('--lines')

886

self.assertFailure('-l', 'spam')

887

stdout = self.run_ok('--lines', '2', '2004')

888

self.assertIn(conv('December\n\nMo Tu We'), stdout)

889

890

def test_option_spacing(self):

891

self.assertFailure('-s')

892

self.assertFailure('--spacing')

893

self.assertFailure('-s', 'spam')

894

stdout = self.run_ok('--spacing', '8', '2004')

895

self.assertIn(b'Su Mo', stdout)

896

897

def test_option_months(self):

898

self.assertFailure('-m')

899

self.assertFailure('--month')

900

self.assertFailure('-m', 'spam')

901

stdout = self.run_ok('--months', '1', '2004')

902

self.assertIn(conv('\nMo Tu We Th Fr Sa Su\n'), stdout)

903

904

def test_option_type(self):

905

self.assertFailure('-t')

906

self.assertFailure('--type')

907

self.assertFailure('-t', 'spam')

908

stdout = self.run_ok('--type', 'text', '2004')

909

self.assertEqual(stdout, conv(result_2004_text))

910

stdout = self.run_ok('--type', 'html', '2004')

911

self.assertEqual(stdout[:6], b'<?xml ')

912

self.assertIn(b'<title>Calendar for 2004</title>', stdout)

913

914

def test_html_output_current_year(self):

915

stdout = self.run_ok('--type', 'html')

916

year = datetime.datetime.now().year

917

self.assertIn(('<title>Calendar for %s</title>' % year).encode(),

918

stdout)

919

self.assertIn(b'<tr><th colspan="7" class="month">January</th></tr>',

920

stdout)

921

922

def test_html_output_year_encoding(self):

923

stdout = self.run_ok('-t', 'html', '--encoding', 'ascii', '2004')

924

self.assertEqual(stdout,

Oz N Tiram

2017-06-06 11:35:59 +0200

[diff] [blame]

925

result_2004_html.format(**default_format).encode('ascii'))

Serhiy Storchaka

2015-02-05 15:14:35 +0200

[diff] [blame]

926

927

def test_html_output_year_css(self):

928

self.assertFailure('-t', 'html', '-c')

929

self.assertFailure('-t', 'html', '--css')

930

stdout = self.run_ok('-t', 'html', '--css', 'custom.css', '2004')

931

self.assertIn(b'<link rel="stylesheet" type="text/css" '

932

b'href="custom.css" />', stdout)

Walter Dörwald

2004-07-02 19:00:09 +0000

[diff] [blame]

933

Fred Drake

2001-12-12 05:38:08 +0000

[diff] [blame]

934

Martin Panter

4eb376c

2016-01-16 06:49:30 +0000

[diff] [blame]

935

class MiscTestCase(unittest.TestCase):

936

def test__all__(self):

Victor Stinner

fbf43f0

2020-08-17 07:20:40 +0200

[diff] [blame]

937

not_exported = {

938

'mdays', 'January', 'February', 'EPOCH', 'MONDAY', 'TUESDAY',

939

'WEDNESDAY', 'THURSDAY', 'FRIDAY', 'SATURDAY', 'SUNDAY',

940

'different_locale', 'c', 'prweek', 'week', 'format',

941

'formatstring', 'main', 'monthlen', 'prevmonth', 'nextmonth'}

942

support.check__all__(self, calendar, not_exported=not_exported)

Martin Panter

4eb376c

2016-01-16 06:49:30 +0000

[diff] [blame]

943

944

Oz N Tiram

2017-06-06 11:35:59 +0200

[diff] [blame]

945

class TestSubClassingCase(unittest.TestCase):

def setUp(self):

class CustomHTMLCal(calendar.HTMLCalendar):

950

cssclasses = [style + " text-nowrap" for style in

951

calendar.HTMLCalendar.cssclasses]

952

cssclasses_weekday_head = ["red", "blue", "green", "lilac",

953

"yellow", "orange", "pink"]

954

cssclass_month_head = "text-center month-head"

955

cssclass_month = "text-center month"

956

cssclass_year = "text-italic "

957

cssclass_year_head = "lead "

958

959

self.cal = CustomHTMLCal()

960

961

def test_formatmonthname(self):

962

self.assertIn('class="text-center month-head"',

963

self.cal.formatmonthname(2017, 5))

964

965

def test_formatmonth(self):

966

self.assertIn('class="text-center month"',

967

self.cal.formatmonth(2017, 5))

968

969

def test_formatweek(self):

970

weeks = self.cal.monthdays2calendar(2017, 5)

971

self.assertIn('class="wed text-nowrap"', self.cal.formatweek(weeks[0]))

972

973

def test_formatweek_head(self):

974

header = self.cal.formatweekheader()

975

for color in self.cal.cssclasses_weekday_head:

976

self.assertIn('<th class="%s">' % color, header)

977

978

def test_format_year(self):

979

self.assertIn(

980

('<table border="0" cellpadding="0" cellspacing="0" class="%s">' %

981

self.cal.cssclass_year), self.cal.formatyear(2017))

982

983

def test_format_year_head(self):

984

self.assertIn('<tr><th colspan="%d" class="%s">%s</th></tr>' % (

985

3, self.cal.cssclass_year_head, 2017), self.cal.formatyear(2017))

986

Fred Drake

2001-12-12 05:38:08 +0000

[diff] [blame]

987

if __name__ == "__main__":

Serhiy Storchaka