FilesExpand file tree

 develop

/

test_custom_widgets.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

94 lines (79 loc) · 2.2 KB

 develop

/

test_custom_widgets.py

Copy path

Top

File metadata and controls

• Code
• Blame

94 lines (79 loc) · 2.2 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89

90

91

92

93

94

import time

import progressbar

import pytest

class CrazyFileTransferSpeed(progressbar.FileTransferSpeed):

"It's bigger between 45 and 80 percent"

def update(self, pbar):

if 45 < pbar.percentage() < 80:

value = progressbar.FileTransferSpeed.update(self, pbar)

return f'Bigger Now {value}'

else:

return progressbar.FileTransferSpeed.update(self, pbar)

def test_crazy_file_transfer_speed_widget():

widgets = [

# CrazyFileTransferSpeed(),

' <<<',

progressbar.Bar(),

'>>> ',

progressbar.Percentage(),

' ',

progressbar.ETA(),

]

p = progressbar.ProgressBar(widgets=widgets, max_value=1000)

# maybe do something

p.start()

for i in range(0, 200, 5):

# do something

time.sleep(0.1)

p.update(i + 1)

p.finish()

def test_variable_widget_widget():

widgets = [

' [',

progressbar.Timer(),

'] ',

progressbar.Bar(),

' (',

progressbar.ETA(),

') ',

progressbar.Variable('loss'),

progressbar.Variable('text'),

progressbar.Variable('error', precision=None),

progressbar.Variable('missing'),

progressbar.Variable('predefined'),

]

p = progressbar.ProgressBar(

widgets=widgets,

max_value=1000,

variables=dict(predefined='predefined'),

)

p.start()

print('time', time, time.sleep)

for i in range(0, 200, 5):

time.sleep(0.1)

p.update(i + 1, loss=0.5, text='spam', error=1)

i += 1

p.update(i, text=None)

i += 1

p.update(i, text=False)

i += 1

p.update(i, text=True, error='a')

with pytest.raises(TypeError):

p.update(i, non_existing_variable='error!')

p.finish()

def test_format_custom_text_widget():

widget = progressbar.FormatCustomText(

'Spam: %(spam).1f kg, eggs: %(eggs)d',

dict(

spam=0.25,

eggs=3,

),

)

bar = progressbar.ProgressBar(

widgets=[

widget,

],

)

for i in bar(range(5)):

widget.update_mapping(eggs=i * 2)

assert widget.mapping['eggs'] == bar.widgets[0].mapping['eggs']