OLD | NEW |
| (Empty) |
1 #!/usr/bin/env python | |
2 # Copyright (c) 2016 The WebRTC project authors. All Rights Reserved. | |
3 # | |
4 # Use of this source code is governed by a BSD-style license | |
5 # that can be found in the LICENSE file in the root of the source | |
6 # tree. An additional intellectual property rights grant can be found | |
7 # in the file PATENTS. All contributing project authors may | |
8 # be found in the AUTHORS file in the root of the source tree. | |
9 | |
10 """Run the tests with | |
11 | |
12 python rtp_analyzer_test.py | |
13 or | |
14 python3 rtp_analyzer_test.py | |
15 """ | |
16 | |
17 import collections | |
18 import unittest | |
19 | |
20 MISSING_NUMPY = False # pylint: disable=invalid-name | |
21 try: | |
22 import numpy | |
23 import rtp_analyzer | |
24 except ImportError: | |
25 MISSING_NUMPY = True | |
26 | |
27 FakePoint = collections.namedtuple("FakePoint", | |
28 ["real_send_time_ms", "absdelay"]) | |
29 | |
30 | |
31 class TestDelay(unittest.TestCase): | |
32 def AssertMaskEqual(self, masked_array, data, mask): | |
33 self.assertEqual(list(masked_array.data), data) | |
34 | |
35 if isinstance(masked_array.mask, numpy.bool_): | |
36 array_mask = masked_array.mask | |
37 else: | |
38 array_mask = list(masked_array.mask) | |
39 self.assertEqual(array_mask, mask) | |
40 | |
41 def testCalculateDelaySimple(self): | |
42 points = [FakePoint(0, 0), FakePoint(1, 0)] | |
43 mask = rtp_analyzer.CalculateDelay(0, 1, 1, points) | |
44 self.AssertMaskEqual(mask, [0, 0], False) | |
45 | |
46 def testCalculateDelayMissing(self): | |
47 points = [FakePoint(0, 0), FakePoint(2, 0)] | |
48 mask = rtp_analyzer.CalculateDelay(0, 2, 1, points) | |
49 self.AssertMaskEqual(mask, [0, -1, 0], [False, True, False]) | |
50 | |
51 def testCalculateDelayBorders(self): | |
52 points = [FakePoint(0, 0), FakePoint(2, 0)] | |
53 mask = rtp_analyzer.CalculateDelay(0, 3, 2, points) | |
54 self.AssertMaskEqual(mask, [0, 0, -1], [False, False, True]) | |
55 | |
56 | |
57 if __name__ == "__main__": | |
58 if MISSING_NUMPY: | |
59 print "Missing numpy, skipping test." | |
60 else: | |
61 unittest.main() | |
OLD | NEW |