|
|
|
@@ -83,3 +83,23 @@ class TestHistogram: |
|
|
|
assert buckets == ( |
|
|
|
(-1.0, 0.8, 0), (-0.19999999999999996, 0.8, 1), (0.6000000000000001, 0.8, 5), (1.4000000000000004, 0.8, 6), |
|
|
|
(2.2, 0.8, 0)) |
|
|
|
|
|
|
|
def test_re_sample_buckets_zero_width(self): |
|
|
|
"""Test zero width bucket when re-sampling.""" |
|
|
|
mocked_input = mock.MagicMock() |
|
|
|
mocked_bucket = mock.MagicMock() |
|
|
|
mocked_bucket.left = 0 |
|
|
|
mocked_bucket.width = 1 |
|
|
|
mocked_bucket.count = 1 |
|
|
|
mocked_bucket2 = mock.MagicMock() |
|
|
|
mocked_bucket2.left = 1 |
|
|
|
mocked_bucket2.width = 0 |
|
|
|
mocked_bucket2.count = 2 |
|
|
|
mocked_input.buckets = [mocked_bucket, mocked_bucket2] |
|
|
|
histogram = hist.HistogramContainer(mocked_input) |
|
|
|
histogram.set_visual_range(max_val=2, min_val=0, bins=3) |
|
|
|
buckets = histogram.buckets() |
|
|
|
assert buckets == ( |
|
|
|
(0.0, 0.6666666666666666, 1), |
|
|
|
(0.6666666666666666, 0.6666666666666666, 3), |
|
|
|
(1.3333333333333333, 0.6666666666666666, 0)) |