infra_libs/ts_mon/common/test/distribution_test.py - Issue 2213143002: Add infra_libs as a bootstrap dependency.

Side by Side Diff: infra_libs/ts_mon/common/test/distribution_test.py

Issue 2213143002: Add infra_libs as a bootstrap dependency. (Closed) Base URL: https://chromium.googlesource.com/infra/infra.git@master

Patch Set: Removed the ugly import hack Created 4 years, 4 months ago

Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.

Jump to:

View unified diff | Download patch

OLD	NEW
	(Empty)
1 # Copyright 2015 The Chromium Authors. All rights reserved.

2 # Use of this source code is governed by a BSD-style license that can be

3 # found in the LICENSE file.

4

5 import unittest

6

7 from infra_libs.ts_mon.common import distribution

8

9

10 class BucketerTestBase(unittest.TestCase):

11 def assertBucketCounts(self, b, expected_total):

12 self.assertEquals(expected_total - 2, b.num_finite_buckets)

13 self.assertEquals(expected_total, b.total_buckets)

14 self.assertEquals(0, b.underflow_bucket)

15 self.assertEquals(expected_total - 1, b.overflow_bucket)

16

17 def assertBoundaries(self, b, expected_finite_upper_boundaries):

18 self.assertEquals(

19 b.num_finite_buckets, len(expected_finite_upper_boundaries))

20

21 expected_boundaries = [(float('-Inf'), 0)]

22

23 previous = 0

24 for value in expected_finite_upper_boundaries:

25 expected_boundaries.append((previous, value))

26 previous = value

27

28 expected_boundaries.append((previous, float('Inf')))

29

30 self.assertEquals(expected_boundaries, list(b.all_bucket_boundaries()))

31 for i, expected in enumerate(expected_boundaries):

32 self.assertEquals(expected, b.bucket_boundaries(i))

33 self.assertEquals(i, b.bucket_for_value(expected[0]))

34 self.assertEquals(i, b.bucket_for_value(expected[0] + 0.5))

35 self.assertEquals(i, b.bucket_for_value(expected[1] - 0.5))

36

37 with self.assertRaises(IndexError):

38 b.bucket_boundaries(-1)

39 with self.assertRaises(IndexError):

40 b.bucket_boundaries(len(expected_boundaries))

41

42

43 class FixedWidthBucketerTest(BucketerTestBase):

44 def test_negative_size(self):

45 with self.assertRaises(ValueError):

46 distribution.FixedWidthBucketer(width=10, num_finite_buckets=-1)

47

48 def test_negative_width(self):

49 with self.assertRaises(ValueError):

50 distribution.FixedWidthBucketer(width=-1, num_finite_buckets=1)

51

52 def test_zero_size(self):

53 b = distribution.FixedWidthBucketer(width=10, num_finite_buckets=0)

54

55 self.assertBucketCounts(b, 2)

56 self.assertBoundaries(b, [])

57

58 def test_one_size(self):

59 b = distribution.FixedWidthBucketer(width=10, num_finite_buckets=1)

60

61 self.assertBucketCounts(b, 3)

62 self.assertBoundaries(b, [10])

63

64

65 class GeometricBucketerTest(BucketerTestBase):

66 def test_negative_size(self):

67 with self.assertRaises(ValueError):

68 distribution.GeometricBucketer(num_finite_buckets=-1)

69

70 def test_small_scale(self):

71 with self.assertRaises(ValueError):

72 distribution.GeometricBucketer(growth_factor=-1)

73 with self.assertRaises(ValueError):

74 distribution.GeometricBucketer(growth_factor=0)

75 with self.assertRaises(ValueError):

76 distribution.GeometricBucketer(growth_factor=1)

77

78 def test_zero_size(self):

79 b = distribution.GeometricBucketer(num_finite_buckets=0)

80

81 self.assertBucketCounts(b, 2)

82 self.assertBoundaries(b, [])

83

84 def test_large_size(self):

85 b = distribution.GeometricBucketer(growth_factor=4, num_finite_buckets=4)

86

87 self.assertBucketCounts(b, 6)

88 self.assertBoundaries(b, [1, 4, 16, 64])

89

90

91 class CustomBucketerTest(BucketerTestBase):

92 def test_boundaries(self):

93 b = distribution.Bucketer(width=10, growth_factor=2, num_finite_buckets=4)

94

95 self.assertBucketCounts(b, 6)

96 self.assertBoundaries(b, [11, 22, 34, 48])

97

98

99 class DistributionTest(unittest.TestCase):

100 def test_add(self):

101 d = distribution.Distribution(distribution.GeometricBucketer())

102 self.assertEqual(0, d.sum)

103 self.assertEqual(0, d.count)

104 self.assertEqual({}, d.buckets)

105

106 d.add(1)

107 d.add(10)

108 d.add(100)

109

110 self.assertEqual(111, d.sum)

111 self.assertEqual(3, d.count)

112 self.assertEqual({2: 1, 6: 1, 11: 1}, d.buckets)

113

114 d.add(50)

115

116 self.assertEqual(161, d.sum)

117 self.assertEqual(4, d.count)

118 self.assertEqual({2: 1, 6: 1, 10: 1, 11: 1}, d.buckets)

119

120 def test_add_on_bucket_boundary(self):

121 d = distribution.Distribution(distribution.FixedWidthBucketer(width=10))

122

123 d.add(10)

124

125 self.assertEqual(10, d.sum)

126 self.assertEqual(1, d.count)

127 self.assertEqual({2: 1}, d.buckets)

128

129 d.add(0)

130

131 self.assertEqual(10, d.sum)

132 self.assertEqual(2, d.count)

133 self.assertEqual({1: 1, 2: 1}, d.buckets)

134

135 def test_underflow_bucket(self):

136 d = distribution.Distribution(distribution.FixedWidthBucketer(width=10))

137

138 d.add(-1)

139

140 self.assertEqual(-1, d.sum)

141 self.assertEqual(1, d.count)

142 self.assertEqual({0: 1}, d.buckets)

143

144 d.add(-1000000)

145

146 self.assertEqual(-1000001, d.sum)

147 self.assertEqual(2, d.count)

148 self.assertEqual({0: 2}, d.buckets)

149

150 def test_overflow_bucket(self):

151 d = distribution.Distribution(

152 distribution.FixedWidthBucketer(width=10, num_finite_buckets=10))

153

154 d.add(100)

155

156 self.assertEqual(100, d.sum)

157 self.assertEqual(1, d.count)

158 self.assertEqual({11: 1}, d.buckets)

159

160 d.add(1000000)

161

162 self.assertEqual(1000100, d.sum)

163 self.assertEqual(2, d.count)

164 self.assertEqual({11: 2}, d.buckets)

OLD	NEW

« bootstrap/deps.pyl ('K') | « infra_libs/ts_mon/common/test/__init__.py ('k') | infra_libs/ts_mon/common/test/errors_test.py » ('j') | no next file with comments »