forked from jopohl/urh
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtest_iq_array.py
114 lines (80 loc) · 5.36 KB
/
test_iq_array.py
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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
import unittest
import numpy as np
from urh.signalprocessing.IQArray import IQArray
class TestIQArray(unittest.TestCase):
def test_index(self):
iq_array = IQArray(np.array([1, 2, 3, 4, 5, 6], dtype=np.uint8))
self.assertEqual(iq_array[0][0], 1)
self.assertEqual(iq_array[0][1], 2)
self.assertEqual(iq_array[1][0], 3)
self.assertEqual(iq_array[1][1], 4)
self.assertEqual(iq_array[2][0], 5)
self.assertEqual(iq_array[2][1], 6)
self.assertEqual(iq_array[1:2][0, 0], 3)
self.assertEqual(iq_array[1:2][0, 1], 4)
self.assertEqual(iq_array[:2][0, 0], 1)
self.assertEqual(iq_array[:2][0, 1], 2)
self.assertEqual(iq_array[:2][1, 0], 3)
self.assertEqual(iq_array[:2][1, 1], 4)
iq_array[0] = np.array([13, 37])
self.assertEqual(iq_array[0][0], 13)
self.assertEqual(iq_array[0][1], 37)
iq_array[0:2] = np.array([42, 42, 47, 11])
self.assertEqual(iq_array[0][0], 42)
self.assertEqual(iq_array[0][1], 42)
self.assertEqual(iq_array[1][0], 47)
self.assertEqual(iq_array[1][1], 11)
def test_conversion_iq16s(self):
iq16s = IQArray(np.array([-128, 0, 0, 127], dtype=np.int8))
self.assertTrue(np.array_equal(iq16s.convert_to(np.int8).flatten(), np.array([-128, 0, 0, 127], dtype=np.int8)))
self.assertTrue(np.array_equal(iq16s.convert_to(np.uint8).flatten(), np.array([0, 128, 128, 255], dtype=np.uint8)))
c32s = iq16s.convert_to(np.int16).flatten()
self.assertTrue(np.array_equal(c32s, np.array([-32768, 0, 0, 32512], dtype=np.int16)), msg=c32s)
c32u = iq16s.convert_to(np.uint16).flatten()
self.assertTrue(np.array_equal(c32u, np.array([0, 32768, 32768, 65280], dtype=np.uint16)), msg=c32u)
c64f = iq16s.convert_to(np.float32).flatten()
self.assertTrue(np.array_equal(c64f, np.array([-1, 0, 0, 0.9921875], dtype=np.float32)), msg=c64f)
def test_conversion_iq16u(self):
iq16u = IQArray(np.array([0, 128, 128, 255], dtype=np.uint8))
self.assertTrue(np.array_equal(iq16u.convert_to(np.uint8).flatten(), np.array([0, 128, 128, 255], dtype=np.uint8)))
iq16s = iq16u.convert_to(np.int8).flatten()
self.assertTrue(np.array_equal(iq16s, np.array([-128, 0, 0, 127], dtype=np.int8)), msg=iq16s)
c32s = iq16u.convert_to(np.int16).flatten()
self.assertTrue(np.array_equal(c32s, np.array([-32768, 0, 0, 32512], dtype=np.int16)), msg=c32s)
c32u = iq16u.convert_to(np.uint16).flatten()
self.assertTrue(np.array_equal(c32u, np.array([0, 32768, 32768, 65280], dtype=np.uint16)), msg=c32u)
c64f = iq16u.convert_to(np.float32).flatten()
self.assertTrue(np.array_equal(c64f, np.array([-1, 0, 0, 0.9921875], dtype=np.float32)), msg=c64f)
def test_conversion_iq32s(self):
iq32s = IQArray(np.array([-32768, 0, 0, 32767], dtype=np.int16))
self.assertTrue(np.array_equal(iq32s.convert_to(np.int16).flatten(), np.array([-32768, 0, 0, 32767], dtype=np.int16)))
iq32u = iq32s.convert_to(np.uint16).flatten()
self.assertTrue(np.array_equal(iq32u, np.array([0, 32768, 32768, 65535], dtype=np.uint16)), msg=iq32u)
iq16s = iq32s.convert_to(np.int8).flatten()
self.assertTrue(np.array_equal(iq16s, np.array([-128, 0, 0, 127], dtype=np.int8)), msg=iq16s)
iq16u = iq32s.convert_to(np.uint8).flatten()
self.assertTrue(np.array_equal(iq16u, np.array([0, 128, 128, 255], dtype=np.uint8)), msg=iq16u)
iq64f = iq32s.convert_to(np.float32).flatten()
self.assertTrue(np.array_equal(iq64f, np.array([-1, 0, 0, 0.9999695], dtype=np.float32)), msg=iq64f)
def test_conversion_iq32u(self):
iq32u = IQArray(np.array([0, 32768, 32768, 65535], dtype=np.uint16))
self.assertTrue(np.array_equal(iq32u.convert_to(np.uint16).flatten(), np.array([0, 32768, 32768, 65535], dtype=np.uint16)))
iq32s = iq32u.convert_to(np.int16).flatten()
self.assertTrue(np.array_equal(iq32s, np.array([-32768, 0, 0, 32767], dtype=np.int16)), msg=iq32s)
iq16s = iq32u.convert_to(np.int8).flatten()
self.assertTrue(np.array_equal(iq16s, np.array([-128, 0, 0, 127], dtype=np.int8)), msg=iq16s)
iq16u = iq32u.convert_to(np.uint8).flatten()
self.assertTrue(np.array_equal(iq16u, np.array([0, 128, 128, 255], dtype=np.uint8)), msg=iq16u)
iq64f = iq32u.convert_to(np.float32).flatten()
self.assertTrue(np.array_equal(iq64f, np.array([-1, 0, 0, 0.9999695], dtype=np.float32)), msg=iq64f)
def test_conversion_iq64f(self):
iq64f = IQArray(np.array([-1, 0, 0, 1], dtype=np.float32))
self.assertTrue(np.array_equal(iq64f.convert_to(np.float32).flatten(), np.array([-1, 0, 0, 1], dtype=np.float32)))
iq16u = iq64f.convert_to(np.uint8).flatten()
self.assertTrue(np.array_equal(iq16u, np.array([0, 127, 127, 254], dtype=np.uint8)), msg=iq16u)
iq16s = iq64f.convert_to(np.int8).flatten()
self.assertTrue(np.array_equal(iq16s, np.array([-127, 0, 0, 127], dtype=np.int8)), msg=iq16s)
iq32s = iq64f.convert_to(np.int16).flatten()
self.assertTrue(np.array_equal(iq32s, np.array([-32767, 0, 0, 32767], dtype=np.int16)), msg=iq32s)
iq32u = iq64f.convert_to(np.uint16).flatten()
self.assertTrue(np.array_equal(iq32u, np.array([0, 32767, 32767, 65534], dtype=np.uint16)), msg=iq32u)