HackRF-Treasure-Chest/Software/Universal Radio Hacker/tests/test_file_operator.py

66 lines
2.8 KiB
Python
Raw Permalink Normal View History

2022-09-22 22:46:47 +02:00
import hashlib
import os
import tarfile
import tempfile
from zipfile import ZipFile
import numpy as np
from PyQt5.QtCore import QDir
from PyQt5.QtTest import QTest
from PyQt5.QtWidgets import QApplication, QFileDialog
from tests.QtTestCase import QtTestCase
from urh.signalprocessing.IQArray import IQArray
from urh.util import FileOperator
class TestFileOperator(QtTestCase):
def test_save_wav(self):
temp_dir = tempfile.gettempdir()
os.chdir(temp_dir)
self.assertFalse(os.path.isfile("test.wav"))
FileOperator.save_data(np.array([1, 2], dtype=np.int16), "test.wav")
self.assertTrue(os.path.isfile("test.wav"))
os.remove("test.wav")
def test_uncompress_archives(self):
temp_dir = tempfile.gettempdir()
os.chdir(temp_dir)
with tarfile.open("test.tar.gz", "w:gz") as tar:
for name in ["1.complex", "2.complex", "3.complex"]:
data = np.ones(10, dtype=np.complex64)
data.tofile(name)
tar.add(name)
with ZipFile('test.zip', 'w') as zip:
for name in ["4.complex", "5.complex"]:
data = np.ones(20, dtype=np.complex64)
data.tofile(name)
zip.write(name)
self.form.add_files(FileOperator.uncompress_archives(["test.tar.gz", "test.zip"], QDir.tempPath()))
self.assertEqual(len(self.form.signal_tab_controller.signal_frames), 5)
tar_md5 = hashlib.md5(open(os.path.join(temp_dir, "test.tar.gz"), 'rb').read()).hexdigest()
self.form.signal_tab_controller.signal_frames[0].signal.iq_array = IQArray(np.ones(5, dtype=np.complex64))
self.form.signal_tab_controller.signal_frames[0].signal.changed = True
self.form.signal_tab_controller.signal_frames[0].ui.btnSaveSignal.click()
tar_md5_after_save = hashlib.md5(open(os.path.join(temp_dir, "test.tar.gz"), 'rb').read()).hexdigest()
self.assertNotEqual(tar_md5, tar_md5_after_save)
zip_md5 = hashlib.md5(open(os.path.join(temp_dir, "test.zip"), 'rb').read()).hexdigest()
self.form.signal_tab_controller.signal_frames[4].signal.iq_array = IQArray(np.ones(5, dtype=np.complex64))
self.form.signal_tab_controller.signal_frames[4].signal.changed = True
self.form.signal_tab_controller.signal_frames[4].ui.btnSaveSignal.click()
zip_md5_after_save = hashlib.md5(open(os.path.join(temp_dir, "test.zip"), 'rb').read()).hexdigest()
self.assertNotEqual(zip_md5, zip_md5_after_save)
def test_get_open_dialog(self):
d1 = FileOperator.get_open_dialog(directory_mode=False)
self.assertEqual(d1.fileMode(), QFileDialog.ExistingFiles)
d2 = FileOperator.get_open_dialog(directory_mode=True)
self.assertEqual(d2.fileMode(), QFileDialog.Directory)