mirror of
https://github.com/jopohl/urh.git
synced 2026-03-10 02:06:46 +01:00
42 lines
1.7 KiB
Python
42 lines
1.7 KiB
Python
from tests.QtTestCase import QtTestCase
|
|
from tests.utils_testing import get_path_for_data_file
|
|
from urh.signalprocessing.ProtocolAnalyzer import ProtocolAnalyzer
|
|
from urh.signalprocessing.Signal import Signal
|
|
|
|
|
|
class TestDemodulations(QtTestCase):
|
|
# Testmethode muss immer mit Präfix test_* starten
|
|
def test_ask(self):
|
|
signal = Signal(get_path_for_data_file("ask.complex"), "ASK-Test")
|
|
signal.modulation_type = 0
|
|
signal.bit_len = 295
|
|
signal.qad_center = -0.1667
|
|
self.assertEqual(signal.num_samples, 13710)
|
|
|
|
proto_analyzer = ProtocolAnalyzer(signal)
|
|
proto_analyzer.get_protocol_from_signal()
|
|
self.assertTrue(proto_analyzer.plain_bits_str[0].startswith("1011001001011011011011011011011011001000000"))
|
|
|
|
def test_fsk(self):
|
|
signal = Signal(get_path_for_data_file("fsk.complex"), "FSK-Test")
|
|
signal.modulation_type = 1
|
|
signal.bit_len = 100
|
|
signal.qad_center = 0
|
|
|
|
proto_analyzer = ProtocolAnalyzer(signal)
|
|
proto_analyzer.get_protocol_from_signal()
|
|
self.assertEqual(proto_analyzer.plain_bits_str[0],
|
|
"101010101010101010101010101010101100011000100110110001100010011011110100110111000001110110011000111011101111011110100100001001111001100110011100110100100011100111010011111100011")
|
|
|
|
def test_psk(self):
|
|
signal = Signal(get_path_for_data_file("psk_gen_noisy.complex"), "PSK-Test")
|
|
signal.modulation_type = 2
|
|
signal.bit_len = 300
|
|
signal.qad_center = 0.0281
|
|
signal.noise_threshold = 0
|
|
signal.tolerance = 10
|
|
|
|
proto_analyzer = ProtocolAnalyzer(signal)
|
|
proto_analyzer.get_protocol_from_signal()
|
|
self.assertEqual(proto_analyzer.plain_bits_str[0], "101100")
|