Ejemplo n.º 1
0
 def test_div_update(self):
     a = TimeSeries([(1, 3), (2, 4), (3, 3)])
     b = TimeSeries([(0, 2), (1, 3), (2, 2), (3, 2), (4, 1)])
     a /= b
     self.assertListEqual(a.points, [(1, 1), (2, 2), (3, 1.5)])
     a /= 0.5
     self.assertListEqual(a.points, [(1, 2), (2, 4), (3, 3)])
Ejemplo n.º 2
0
 def test_pow_update(self):
     a = TimeSeries([(1, 3), (2, 3), (3, 3)])
     b = TimeSeries([(0, 2), (1, 3), (2, 2), (3, 1), (4, 1)])
     a **= b
     self.assertListEqual(a.points, [(1, 27), (2, 9), (3, 3)])
     a **= 2
     self.assertListEqual(a.points, [(1, 729), (2, 81), (3, 9)])
Ejemplo n.º 3
0
 def test_mul_update(self):
     a = TimeSeries([(1, 3), (2, 3), (3, 3)])
     b = TimeSeries([(0, 2), (1, 3), (2, 2), (3, 2), (4, 1)])
     a *= b
     self.assertListEqual(a.points, [(1, 9), (2, 6), (3, 6)])
     a *= 2
     self.assertListEqual(a.points, [(1, 18), (2, 12), (3, 12)])
Ejemplo n.º 4
0
 def test_sub_update(self):
     a = TimeSeries([(1, 3), (2, 3), (3, 3)])
     b = TimeSeries([(0, 1), (1, 1), (2, 1), (3, 1), (4, 1)])
     a -= b
     self.assertListEqual(a.points, [(1, 2), (2, 2), (3, 2)])
     a -= 1
     self.assertListEqual(a.points, [(1, 1), (2, 1), (3, 1)])
Ejemplo n.º 5
0
 def test_pow(self):
     a = TimeSeries([(1, 3), (2, 3), (3, 3)])
     b = TimeSeries([(0, 2), (1, 3), (2, 2), (3, 1), (4, 1)])
     c = a**b
     self.assertTrue(isinstance(c, TimeSeries))
     self.assertListEqual(c.points, [(1, 27), (2, 9), (3, 3)])
     c = c**2
     self.assertTrue(isinstance(c, TimeSeries))
     self.assertListEqual(c.points, [(1, 729), (2, 81), (3, 9)])
Ejemplo n.º 6
0
 def test_div(self):
     a = TimeSeries([(1, 3), (2, 4), (3, 3)])
     b = TimeSeries([(0, 2), (1, 3), (2, 2), (3, 2), (4, 1)])
     c = a / b
     self.assertTrue(isinstance(c, TimeSeries))
     self.assertListEqual(c.points, [(1, 1), (2, 2), (3, 1.5)])
     c = c / 2
     self.assertTrue(isinstance(c, TimeSeries))
     self.assertListEqual(c.points, [(1, 0.5), (2, 1), (3, 0.75)])
Ejemplo n.º 7
0
 def test_mul(self):
     a = TimeSeries([(1, 3), (2, 3), (3, 3)])
     b = TimeSeries([(0, 2), (1, 3), (2, 2), (3, 2), (4, 1)])
     c = a * b
     self.assertTrue(isinstance(c, TimeSeries))
     self.assertListEqual(c.points, [(1, 9), (2, 6), (3, 6)])
     c = c * 2
     self.assertTrue(isinstance(c, TimeSeries))
     self.assertListEqual(c.points, [(1, 18), (2, 12), (3, 12)])
Ejemplo n.º 8
0
 def test_time_series_repr(self):
     a = TimeSeries([(10000, 500), (20000, 1234), (30000, 12345678)])
     lines = repr(a).split('\n')
     self.assertEqual(lines[0], '                        Value')
     self.assertTrue(
         re.match(
             r'^1970-[0-9]{2}-[0-9]{2} [0-9]{2}:[0-9]{2}:10       500$',
             lines[1]))
     self.assertTrue(
         re.match(
             r'^1970-[0-9]{2}-[0-9]{2} [0-9]{2}:[0-9]{2}:20      1234$',
             lines[2]))
     self.assertTrue(
         re.match(
             r'^1970-[0-9]{2}-[0-9]{2} [0-9]{2}:[0-9]{2}:30  12345678$',
             lines[3]))
     self.assertEqual(str(TimeSeries([])), 'TimeSeries([])')
Ejemplo n.º 9
0
 def test_map_return_type(self):
     series = TimeSeries([ (1, 2), (3, 4), (5, 6) ])
     double = series.map(lambda y: y * 2)
     self.assertTrue(isinstance(double, TimeSeries))
     self.assertListEqual([ (1, 4), (3, 8), (5, 12) ], double.points)
Ejemplo n.º 10
0
 def test_round(self):
     a = TimeSeries([ (1, -0.3), (2, 3.3), (3, 1.1) ])
     a = a.round()
     self.assertTrue(isinstance(a, TimeSeries))
     self.assertListEqual(a.values, [ 0, 3, 1 ])
Ejemplo n.º 11
0
 def test_dict_init(self):
     series = TimeSeries({1: 2, 3: 4, 5: 6})
     self.assertListEqual(series.timestamps, [1, 3, 5])
     self.assertListEqual(series.values, [2, 4, 6])
     self.assertEqual(len(series), 3)
Ejemplo n.º 12
0
 def test_tuple_list_init(self):
     series = TimeSeries([(1, 2), (3, 4), (5, 6)])
     self.assertListEqual(series.timestamps, [1, 3, 5])
     self.assertListEqual(series.values, [2, 4, 6])
     self.assertEqual(len(series), 3)
Ejemplo n.º 13
0
 def test_round(self):
     a = TimeSeries([(1, -0.3), (2, 3.3), (3, 1.1)])
     a = a.round()
     self.assertTrue(isinstance(a, TimeSeries))
     self.assertListEqual(a.values, [0, 3, 1])
Ejemplo n.º 14
0
 def test_abs(self):
     a = TimeSeries([(1, -3), (2, 3.3), (3, -5)])
     a = abs(a)
     self.assertTrue(isinstance(a, TimeSeries))
     self.assertListEqual(a.values, [3, 3.3, 5])