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

Serhiy Storchaka

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

[diff] [blame]

12

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]

21

Serhiy Storchaka

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

[diff] [blame]

22

result_2004_text = """\

Thomas Wouters

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

[diff] [blame]

23

2004

24

25

January February March

26

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

27

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

28

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

29

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

30

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

31

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

32

33

April May June

34

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

35

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

36

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

37

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

38

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

39

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

40

31

41

42

July August September

43

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

44

1 2 3 4 1 1 2 3 4 5

45

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

46

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

47

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

48

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

49

30 31

50

51

October November December

52

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

53

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

54

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

55

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

56

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

57

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

58

"""

59

Oz N Tiram

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

[diff] [blame]

60

61

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

62

Serhiy Storchaka

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

[diff] [blame]

63

result_2004_html = """\

Oz N Tiram

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

[diff] [blame]

64

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

Thomas Wouters

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

[diff] [blame]

65

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

66

<html>

67

<head>

Oz N Tiram

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

[diff] [blame]

68

Thomas Wouters

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

[diff] [blame]

69

Thomas Wouters

47b49bf

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

[diff] [blame]

70

<title>Calendar for 2004</title>

Thomas Wouters

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

[diff] [blame]

71

</head>

72

<body>

Oz N Tiram

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

[diff] [blame]

73

74

75

<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]

83

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

84

<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]

92

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

93

<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]

101

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

102

<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]

110

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

111

Thomas Wouters

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

[diff] [blame]

</table>

Oz N Tiram

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

[diff] [blame]

120

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

121

Thomas Wouters

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

[diff] [blame]

</table>

Oz N Tiram

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

[diff] [blame]

129

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

130

Thomas Wouters

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

[diff] [blame]

</table>

Oz N Tiram

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

[diff] [blame]

138

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

139

<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]

148

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

149

<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]

157

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

158

<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]

166

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

167

<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]

175

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

176

<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]

188

result_2004_days = [

189

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

190

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

191

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

192

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

193

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

194

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

195

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

196

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

197

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

198

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

199

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

200

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

201

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

202

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

203

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

204

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

205

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

206

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

207

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

208

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

209

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

210

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

211

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

212

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

213

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

214

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

215

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

216

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

217

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

218

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

219

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

220

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

221

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

222

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

223

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

224

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

225

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

226

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

227

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

228

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

229

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

230

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

231

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

232

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

233

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

234

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

235

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

236

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

237

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

238

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

239

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

240

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

241

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

242

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

243

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

244

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

245

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

246

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

247

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

248

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

249

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

250

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

251

]

252

253

result_2004_dates = \

254

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

255

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

256

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

257

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

258

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

259

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

260

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

261

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

262

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

263

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

264

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

265

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

266

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

267

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

268

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

269

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

270

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

271

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

272

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

273

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

274

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

275

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

276

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

277

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

278

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

279

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

280

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

281

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

282

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

283

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

284

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

285

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

286

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

287

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

288

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

289

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

290

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

291

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

292

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

293

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

294

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

295

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

296

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

297

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

298

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

299

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

300

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

301

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

302

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

303

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

304

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

305

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

306

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

307

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

308

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

309

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

310

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

311

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

312

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

313

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

314

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

315

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

316

Thomas Wouters

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

[diff] [blame]

317

318

class OutputTestCase(unittest.TestCase):

319

def normalize_calendar(self, s):

Mark Dickinson

934896d

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

[diff] [blame]

320

# Filters out locale dependent strings

Thomas Wouters

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

[diff] [blame]

321

def neitherspacenordigit(c):

322

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

323

324

lines = []

Ezio Melotti

d8b509b

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

[diff] [blame]

325

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

Thomas Wouters

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

[diff] [blame]

326

# Drop texts, as they are locale dependent

327

if line and not filter(neitherspacenordigit, line):

lines.append(line)

return lines

R David Murray

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

[diff] [blame]

331

def check_htmlcalendar_encoding(self, req, res):

332

cal = calendar.HTMLCalendar()

Oz N Tiram

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

[diff] [blame]

333

format_ = default_format.copy()

334

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

335

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

R David Murray

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

[diff] [blame]

336

self.assertEqual(

Oz N Tiram

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

[diff] [blame]

337

output,

338

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

R David Murray

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

[diff] [blame]

339

)

340

Thomas Wouters

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

[diff] [blame]

341

def test_output(self):

342

self.assertEqual(

343

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

344

self.normalize_calendar(result_2004_text)

345

)

346

347

def test_output_textcalendar(self):

348

self.assertEqual(

Serhiy Storchaka

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

[diff] [blame]

349

calendar.TextCalendar().formatyear(2004),

350

result_2004_text

Thomas Wouters

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

[diff] [blame]

351

)

352

R David Murray

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

[diff] [blame]

353

def test_output_htmlcalendar_encoding_ascii(self):

354

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

355

356

def test_output_htmlcalendar_encoding_utf8(self):

357

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

358

359

def test_output_htmlcalendar_encoding_default(self):

360

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

361

362

def test_yeardatescalendar(self):

363

def shrink(cal):

R David Murray

bb75678

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

[diff] [blame]

364

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

365

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

366

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

Thomas Wouters

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

[diff] [blame]

367

self.assertEqual(

R David Murray

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

[diff] [blame]

368

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

result_2004_dates

)

def test_yeardayscalendar(self):

373

self.assertEqual(

374

calendar.Calendar().yeardayscalendar(2004),

result_2004_days

)

def test_formatweekheader_short(self):

379

self.assertEqual(

380

calendar.TextCalendar().formatweekheader(2),

381

'Mo Tu We Th Fr Sa Su'

382

)

383

384

def test_formatweekheader_long(self):

385

self.assertEqual(

386

calendar.TextCalendar().formatweekheader(9),

387

' Monday Tuesday Wednesday Thursday '

388

' Friday Saturday Sunday '

389

)

390

391

def test_formatmonth(self):

392

self.assertEqual(

Serhiy Storchaka

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

[diff] [blame]

393

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

394

result_2004_01_text

R David Murray

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

[diff] [blame]

395

)

396

397

def test_formatmonthname_with_year(self):

398

self.assertEqual(

399

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

400

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

401

)

402

403

def test_formatmonthname_without_year(self):

404

self.assertEqual(

405

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

406

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

Thomas Wouters

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

[diff] [blame]

407

)

408

Ezio Melotti

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

[diff] [blame]

409

def test_prweek(self):

410

with support.captured_stdout() as out:

411

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

412

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

Serhiy Storchaka

0595ed2

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

[diff] [blame]

413

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

Ezio Melotti

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

[diff] [blame]

414

415

def test_prmonth(self):

416

with support.captured_stdout() as out:

417

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

Serhiy Storchaka

7ff51bd

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

[diff] [blame]

418

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

Ezio Melotti

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

[diff] [blame]

419

420

def test_pryear(self):

421

with support.captured_stdout() as out:

422

calendar.TextCalendar().pryear(2004)

Serhiy Storchaka

0595ed2

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

[diff] [blame]

423

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

Ezio Melotti

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

[diff] [blame]

424

425

def test_format(self):

426

with support.captured_stdout() as out:

427

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

428

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

Thomas Wouters

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

[diff] [blame]

429

Fred Drake

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

[diff] [blame]

430

class CalendarTestCase(unittest.TestCase):

431

def test_isleap(self):

432

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

433

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

434

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

435

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

436

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

437

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

438

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

439

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

440

441

def test_setfirstweekday(self):

Guido van Rossum

47b9ff6

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

[diff] [blame]

442

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

Fred Drake

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

[diff] [blame]

443

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

444

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

445

orig = calendar.firstweekday()

446

calendar.setfirstweekday(calendar.SUNDAY)

447

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

448

calendar.setfirstweekday(calendar.MONDAY)

449

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

450

calendar.setfirstweekday(orig)

451

R David Murray

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

[diff] [blame]

452

def test_illegal_weekday_reported(self):

453

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

454

calendar.setfirstweekday(123)

455

456

def test_enumerate_weekdays(self):

Skip Montanaro

693c6c4

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

[diff] [blame]

457

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

458

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

459

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

Fred Drake

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

[diff] [blame]

460

Tim Peters

0c2c8e7

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

[diff] [blame]

461

def test_days(self):

462

for attr in "day_name", "day_abbr":

463

value = getattr(calendar, attr)

464

self.assertEqual(len(value), 7)

465

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

466

# ensure they're all unique

Tim Peters

bbc0d44

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

[diff] [blame]

467

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

468

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

469

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

Tim Peters

0c2c8e7

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

[diff] [blame]

470

471

def test_months(self):

472

for attr in "month_name", "month_abbr":

473

value = getattr(calendar, attr)

474

self.assertEqual(len(value), 13)

475

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

476

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

477

# ensure they're all unique

Tim Peters

bbc0d44

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

[diff] [blame]

478

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

479

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

480

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

Tim Peters

0c2c8e7

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

[diff] [blame]

481

R David Murray

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

[diff] [blame]

482

def test_locale_calendars(self):

Georg Brandl

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

[diff] [blame]

483

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

484

# (it is still not thread-safe though)

Georg Brandl

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

[diff] [blame]

485

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

Georg Brandl

9450cc0

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

[diff] [blame]

486

try:

Serhiy Storchaka

f9fcdb7

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

[diff] [blame]

487

cal = calendar.LocaleTextCalendar(locale='')

488

local_weekday = cal.formatweekday(1, 10)

489

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

Georg Brandl

9450cc0

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

[diff] [blame]

490

except locale.Error:

491

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

Serhiy Storchaka

f9fcdb7

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

[diff] [blame]

492

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

493

self.assertIsInstance(local_weekday, str)

494

self.assertIsInstance(local_month, str)

495

self.assertEqual(len(local_weekday), 10)

496

self.assertGreaterEqual(len(local_month), 10)

497

cal = calendar.LocaleHTMLCalendar(locale='')

498

local_weekday = cal.formatweekday(1)

499

local_month = cal.formatmonthname(2010, 10)

500

self.assertIsInstance(local_weekday, str)

501

self.assertIsInstance(local_month, str)

Georg Brandl

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

[diff] [blame]

502

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

Ezio Melotti

b3aedd4

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

[diff] [blame]

503

self.assertEqual(old_october, new_october)

Georg Brandl

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

[diff] [blame]

504

Ezio Melotti

85710a4

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

[diff] [blame]

505

def test_itermonthdates(self):

506

# ensure itermonthdates doesn't overflow after datetime.MAXYEAR

507

# see #15421

508

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

509

Alexander Belopolsky

957b756

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

[diff] [blame]

510

def test_itermonthdays(self):

511

for firstweekday in range(7):

512

cal = calendar.Calendar(firstweekday)

513

# Test the extremes, see #28253 and #26650

514

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

515

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

516

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

517

# Test a short month

518

cal = calendar.Calendar(firstweekday=3)

519

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

520

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

521

522

def test_itermonthdays2(self):

523

for firstweekday in range(7):

524

cal = calendar.Calendar(firstweekday)

525

# Test the extremes, see #28253 and #26650

526

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

527

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

528

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

529

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

530

Walter Dörwald

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

[diff] [blame]

531

532

class MonthCalendarTestCase(unittest.TestCase):

533

def setUp(self):

534

self.oldfirstweekday = calendar.firstweekday()

535

calendar.setfirstweekday(self.firstweekday)

536

537

def tearDown(self):

538

calendar.setfirstweekday(self.oldfirstweekday)

539

540

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

541

cal = calendar.monthcalendar(year, month)

542

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

Guido van Rossum

805365e

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

[diff] [blame]

543

for i in range(len(weeks)):

Walter Dörwald

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

[diff] [blame]

544

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

545

546

547

class MondayTestCase(MonthCalendarTestCase):

548

firstweekday = calendar.MONDAY

549

550

def test_february(self):

Thomas Wouters

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

[diff] [blame]

551

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

Walter Dörwald

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

[diff] [blame]

552

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

553

Thomas Wouters

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

[diff] [blame]

554

# 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]

555

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

556

Thomas Wouters

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

[diff] [blame]

557

# 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]

558

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

559

Thomas Wouters

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

[diff] [blame]

560

# 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]

561

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

562

Thomas Wouters

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

[diff] [blame]

563

# 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]

564

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

565

Thomas Wouters

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

[diff] [blame]

566

# 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]

567

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

568

569

def test_april(self):

Thomas Wouters

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

[diff] [blame]

570

# 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]

571

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

572

Thomas Wouters

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

[diff] [blame]

573

# 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]

574

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

575

Thomas Wouters

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

[diff] [blame]

576

# 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]

577

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

578

Thomas Wouters

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

[diff] [blame]

579

# 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]

580

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

581

Thomas Wouters

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

[diff] [blame]

582

# 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]

583

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

584

585

def test_december(self):

Thomas Wouters

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

[diff] [blame]

586

# 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]

587

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

588

Thomas Wouters

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

[diff] [blame]

589

# 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]

590

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

591

Thomas Wouters

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

[diff] [blame]

592

# 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]

593

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

594

Thomas Wouters

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

[diff] [blame]

595

# 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]

596

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

597

Thomas Wouters

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

[diff] [blame]

598

# 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]

599

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

600

Thomas Wouters

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

[diff] [blame]

601

# 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]

602

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

603

604

Walter Dörwald

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

[diff] [blame]

605

class SundayTestCase(MonthCalendarTestCase):

606

firstweekday = calendar.SUNDAY

607

608

def test_february(self):

Thomas Wouters

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

[diff] [blame]

609

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

Walter Dörwald

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

[diff] [blame]

610

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

611

Thomas Wouters

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

[diff] [blame]

612

# 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]

613

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

614

Thomas Wouters

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

[diff] [blame]

615

# 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]

616

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

617

Thomas Wouters

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

[diff] [blame]

618

# 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]

619

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

620

Thomas Wouters

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

[diff] [blame]

621

# 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]

622

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

623

Thomas Wouters

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

[diff] [blame]

624

# 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]

625

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

626

627

def test_april(self):

Thomas Wouters

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

[diff] [blame]

628

# 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]

629

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

630

Thomas Wouters

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

[diff] [blame]

631

# 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]

632

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

633

Thomas Wouters

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

[diff] [blame]

634

# 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]

635

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

636

Thomas Wouters

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

[diff] [blame]

637

# 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]

638

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

639

Thomas Wouters

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

[diff] [blame]

640

# 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]

641

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

642

643

def test_december(self):

Thomas Wouters

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

[diff] [blame]

644

# 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]

645

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

646

Thomas Wouters

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

[diff] [blame]

647

# 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]

648

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

649

Thomas Wouters

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

[diff] [blame]

650

# 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]

651

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

652

Thomas Wouters

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

[diff] [blame]

653

# 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]

654

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

655

Thomas Wouters

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

[diff] [blame]

656

# 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]

657

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

658

Thomas Wouters

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

[diff] [blame]

659

# 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]

660

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

661

Alexander Belopolsky

43ca710

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

[diff] [blame]

662

class TimegmTestCase(unittest.TestCase):

663

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

664

1234567890, 1262304000, 1275785153,]

665

def test_timegm(self):

666

for secs in self.TIMESTAMPS:

667

tuple = time.gmtime(secs)

668

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

Walter Dörwald

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

[diff] [blame]

669

Brett Cannon

cfae36f

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

[diff] [blame]

670

class MonthRangeTestCase(unittest.TestCase):

671

def test_january(self):

672

# Tests valid lower boundary case.

673

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

674

675

def test_february_leap(self):

676

# Tests February during leap year.

677

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

678

679

def test_february_nonleap(self):

680

# Tests February in non-leap year.

681

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

682

683

def test_december(self):

684

# Tests valid upper boundary case.

685

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

686

687

def test_zeroth_month(self):

688

# Tests low invalid boundary case.

689

with self.assertRaises(calendar.IllegalMonthError):

690

calendar.monthrange(2004, 0)

691

692

def test_thirteenth_month(self):

693

# Tests high invalid boundary case.

694

with self.assertRaises(calendar.IllegalMonthError):

695

calendar.monthrange(2004, 13)

696

R David Murray

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

[diff] [blame]

697

def test_illegal_month_reported(self):

698

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

699

calendar.monthrange(2004, 65)

700

R. David Murray

1daaf9e

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

[diff] [blame]

701

class LeapdaysTestCase(unittest.TestCase):

702

def test_no_range(self):

703

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

704

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

705

706

def test_no_leapdays(self):

707

# test when no leap years in range

708

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

709

710

def test_no_leapdays_upper_boundary(self):

711

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

712

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

713

714

def test_one_leapday_lower_boundary(self):

715

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

716

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

717

718

def test_several_leapyears_in_range(self):

719

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

720

Brett Cannon

cfae36f

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

[diff] [blame]

721

Serhiy Storchaka

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

[diff] [blame]

722

def conv(s):

723

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

Senthil Kumaran

962fed9

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

[diff] [blame]

724

Serhiy Storchaka

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

[diff] [blame]

725

class CommandLineTestCase(unittest.TestCase):

726

def run_ok(self, *args):

727

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

728

729

def assertFailure(self, *args):

730

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

Serhiy Storchaka

9785261

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

[diff] [blame]

731

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

Serhiy Storchaka

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

[diff] [blame]

732

self.assertEqual(rc, 2)

733

734

def test_help(self):

735

stdout = self.run_ok('-h')

Serhiy Storchaka

9785261

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

[diff] [blame]

736

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

Serhiy Storchaka

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

[diff] [blame]

737

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

738

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

739

740

def test_illegal_arguments(self):

741

self.assertFailure('-z')

Serhiy Storchaka

9785261

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

[diff] [blame]

742

self.assertFailure('spam')

743

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

Serhiy Storchaka

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

[diff] [blame]

744

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

745

746

def test_output_current_year(self):

747

stdout = self.run_ok()

748

year = datetime.datetime.now().year

749

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

750

self.assertIn(b'January', stdout)

751

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

752

753

def test_output_year(self):

754

stdout = self.run_ok('2004')

755

self.assertEqual(stdout, conv(result_2004_text))

756

757

def test_output_month(self):

758

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

759

self.assertEqual(stdout, conv(result_2004_01_text))

760

761

def test_option_encoding(self):

762

self.assertFailure('-e')

763

self.assertFailure('--encoding')

764

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

765

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

766

767

def test_option_locale(self):

768

self.assertFailure('-L')

769

self.assertFailure('--locale')

770

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

771

lang, enc = locale.getdefaultlocale()

lang = lang or 'C'

enc = enc or 'UTF-8'

try:

oldlocale = locale.getlocale(locale.LC_TIME)

776

try:

777

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

778

finally:

779

locale.setlocale(locale.LC_TIME, oldlocale)

780

except (locale.Error, ValueError):

781

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

782

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

783

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

784

785

def test_option_width(self):

786

self.assertFailure('-w')

787

self.assertFailure('--width')

788

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

789

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

790

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

791

792

def test_option_lines(self):

793

self.assertFailure('-l')

794

self.assertFailure('--lines')

795

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

796

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

797

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

798

799

def test_option_spacing(self):

800

self.assertFailure('-s')

801

self.assertFailure('--spacing')

802

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

803

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

804

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

805

806

def test_option_months(self):

807

self.assertFailure('-m')

808

self.assertFailure('--month')

809

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

810

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

811

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

812

813

def test_option_type(self):

814

self.assertFailure('-t')

815

self.assertFailure('--type')

816

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

817

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

818

self.assertEqual(stdout, conv(result_2004_text))

819

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

820

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

821

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

822

823

def test_html_output_current_year(self):

824

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

825

year = datetime.datetime.now().year

826

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

827

stdout)

828

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

829

stdout)

830

831

def test_html_output_year_encoding(self):

832

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

833

self.assertEqual(stdout,

Oz N Tiram

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

[diff] [blame]

834

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

Serhiy Storchaka

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

[diff] [blame]

835

836

def test_html_output_year_css(self):

837

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

838

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

839

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

840

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

841

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

Walter Dörwald

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

[diff] [blame]

842

Fred Drake

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

[diff] [blame]

843

Martin Panter

4eb376c

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

[diff] [blame]

844

class MiscTestCase(unittest.TestCase):

845

def test__all__(self):

Serhiy Storchaka

2c1d3e3

2016-01-16 11:05:11 +0200

[diff] [blame]

846

blacklist = {'mdays', 'January', 'February', 'EPOCH',

Martin Panter

4eb376c

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

[diff] [blame]

847

'MONDAY', 'TUESDAY', 'WEDNESDAY', 'THURSDAY', 'FRIDAY',

848

'SATURDAY', 'SUNDAY', 'different_locale', 'c',

849

'prweek', 'week', 'format', 'formatstring', 'main'}

850

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

851

852

Oz N Tiram

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

[diff] [blame]

853

class TestSubClassingCase(unittest.TestCase):

def setUp(self):

class CustomHTMLCal(calendar.HTMLCalendar):

858

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

859

calendar.HTMLCalendar.cssclasses]

860

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

861

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

862

cssclass_month_head = "text-center month-head"

863

cssclass_month = "text-center month"

864

cssclass_year = "text-italic "

865

cssclass_year_head = "lead "

866

867

self.cal = CustomHTMLCal()

868

869

def test_formatmonthname(self):

870

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

871

self.cal.formatmonthname(2017, 5))

872

873

def test_formatmonth(self):

874

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

875

self.cal.formatmonth(2017, 5))

876

877

def test_formatweek(self):

878

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

879

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

880

881

def test_formatweek_head(self):

882

header = self.cal.formatweekheader()

883

for color in self.cal.cssclasses_weekday_head:

884

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

885

886

def test_format_year(self):

887

self.assertIn(

888

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

889

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

890

891

def test_format_year_head(self):

892

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

893

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

894

Fred Drake

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

[diff] [blame]

895

if __name__ == "__main__":

Serhiy Storchaka