mirror of
https://github.com/jopohl/urh.git
synced 2026-03-09 09:46:48 +01:00
74dd0bcf2a
* refactor signal file loading * remove "alread quad demod" question * add handling for two channel wavs * add unittests for loading wavs
125 lines
4.9 KiB
Python
125 lines
4.9 KiB
Python
import os
|
|
import tempfile
|
|
import wave
|
|
|
|
import numpy as np
|
|
from PyQt5.QtCore import QTimer
|
|
from PyQt5.QtWidgets import QApplication
|
|
|
|
from tests.QtTestCase import QtTestCase
|
|
from urh.controller.CSVImportDialogController import CSVImportDialogController
|
|
from urh.controller.OptionsController import OptionsController
|
|
|
|
|
|
class TestMaincontrollerGUI(QtTestCase):
|
|
def test_open_recent_file(self):
|
|
# Ensure we have at least one recent action
|
|
self.add_signal_to_form("esaver.complex")
|
|
self.assertEqual(len(self.form.signal_tab_controller.signal_frames), 1)
|
|
|
|
self.form.recentFileActionList[0].trigger()
|
|
self.assertEqual(len(self.form.signal_tab_controller.signal_frames), 2)
|
|
|
|
def test_open_rect_directory(self):
|
|
test_dir = os.path.join(tempfile.gettempdir(), "project_test")
|
|
try:
|
|
os.mkdir(test_dir)
|
|
except OSError:
|
|
pass
|
|
|
|
self.form.project_manager.set_project_folder(test_dir, ask_for_new_project=False)
|
|
|
|
self.assertIn("project_test", self.form.recentFileActionList[0].text())
|
|
|
|
def test_update_decodings(self):
|
|
self.form.update_decodings()
|
|
self.assertTrue(True)
|
|
|
|
def test_options_changed(self):
|
|
self.add_signal_to_form("esaver.complex")
|
|
self.form.on_options_changed({"show_pause_as_time": True, "default_view": 2})
|
|
QApplication.instance().processEvents()
|
|
self.assertEqual(self.form.signal_tab_controller.signal_frames[0].ui.cbProtoView.currentIndex(), 2)
|
|
|
|
def test_open_plain_bits(self):
|
|
bits = ["1010111000110001010101010101",
|
|
"1010111110010010101",
|
|
"1111010100101010101010101010"]
|
|
|
|
filename = os.path.join(tempfile.gettempdir(), "test_plain_bits.txt")
|
|
with open(filename, "w") as f:
|
|
f.write(os.linesep.join(bits))
|
|
|
|
self.form.compare_frame_controller.ui.cbProtoView.setCurrentIndex(0)
|
|
self.wait_before_new_file()
|
|
self.form.add_files([filename])
|
|
|
|
for i, bit_seq in enumerate(bits):
|
|
table_data = "".join(map(str, self.form.compare_frame_controller.protocol_model.display_data[i]))
|
|
self.assertEqual(bit_seq, table_data)
|
|
|
|
def test_open_options_dialog(self):
|
|
self.form.show_options_dialog_specific_tab(1)
|
|
w = next((w for w in QApplication.topLevelWidgets() if isinstance(w, OptionsController)), None) # type: OptionsController
|
|
self.assertIsNotNone(w)
|
|
self.assertEqual(w.ui.tabWidget.currentIndex(), 1)
|
|
w.close()
|
|
|
|
def __accept_csv_dialog(self):
|
|
w = next((w for w in QApplication.topLevelWidgets() if isinstance(w, CSVImportDialogController)), None)
|
|
w.accept()
|
|
|
|
def test_import_csv(self):
|
|
timer = QTimer()
|
|
timer.setInterval(10)
|
|
timer.setSingleShot(True)
|
|
timer.timeout.connect(self.__accept_csv_dialog)
|
|
|
|
self.assertEqual(self.form.signal_tab_controller.num_frames, 0)
|
|
timer.start()
|
|
self.form.add_files([self.get_path_for_filename("csvtest.csv")])
|
|
|
|
self.assertEqual(self.form.signal_tab_controller.signal_frames[0].signal.num_samples, 100)
|
|
self.assertTrue(os.path.isfile(self.get_path_for_filename("csvtest.complex")))
|
|
timer.start()
|
|
self.form.add_files([self.get_path_for_filename("csvtest.csv")])
|
|
|
|
self.assertEqual(self.form.signal_tab_controller.num_frames, 2)
|
|
self.assertTrue(os.path.isfile(self.get_path_for_filename("csvtest_1.complex")))
|
|
|
|
os.remove(self.get_path_for_filename("csvtest.complex"))
|
|
os.remove(self.get_path_for_filename("csvtest_1.complex"))
|
|
|
|
|
|
def test_load_single_channel_wav(self):
|
|
filename = os.path.join(tempfile.gettempdir(), "test_single_channel.wav")
|
|
f = wave.open(filename, "w")
|
|
f.setnchannels(1)
|
|
f.setsampwidth(1)
|
|
f.setframerate(2e6)
|
|
f.writeframes(np.array([1, 2, 3, 4], dtype=np.uint8))
|
|
f.close()
|
|
|
|
self.add_signal_to_form(filename)
|
|
sig_frame = self.form.signal_tab_controller.signal_frames[0]
|
|
self.assertEqual(sig_frame.signal.sample_rate, 2e6)
|
|
self.assertEqual(sig_frame.signal.num_samples, 4)
|
|
self.assertNotEqual(sig_frame.signal.data.real.sum(), 0)
|
|
self.assertEqual(sig_frame.signal.data.imag.sum(), 0)
|
|
|
|
def test_load_stereo_wav(self):
|
|
filename = os.path.join(tempfile.gettempdir(), "test_stereo.wav")
|
|
f = wave.open(filename, "w")
|
|
f.setnchannels(2)
|
|
f.setsampwidth(2)
|
|
f.setframerate(10e6)
|
|
f.writeframes(np.array([23556, 2132, 111, 11111, 3455, 2324], dtype=np.int16))
|
|
f.close()
|
|
|
|
self.add_signal_to_form(filename)
|
|
sig_frame = self.form.signal_tab_controller.signal_frames[0]
|
|
self.assertEqual(sig_frame.signal.sample_rate, 10e6)
|
|
self.assertEqual(sig_frame.signal.num_samples, 3)
|
|
self.assertNotEqual(sig_frame.signal.data.real.sum(), 0)
|
|
self.assertNotEqual(sig_frame.signal.data.imag.sum(), 0)
|