|
- """测试逻辑分析仪 CSV/BIN 波形解析。"""
-
- from __future__ import annotations
-
- import struct
- import tempfile
- import unittest
- from pathlib import Path
-
- from logic_waveform import (
- WaveformChannel,
- build_frequency_curve,
- load_bin_waveform,
- load_csv_waveform,
- parse_channel_selection,
- parse_marker_times_ms,
- )
- from parse_logic_csv import CsvParseError
-
-
- class LogicWaveformTest(unittest.TestCase):
- """验证两种导出格式的关键解析结果。"""
-
- def test_load_csv_waveform(self) -> None:
- """CSV 应识别采样信息、数字时间列和通道跳变。"""
- content = """; Sample rate: 20 MHz
- ; Sample count: 10
- SystemTime,Time(s),Channel 0,Channel 1
- '2026-08-14 10:00:00,0.0,0,1
- '2026-08-14 10:00:00,0.1,1,1
- '2026-08-14 10:00:00,0.2,1,0
- """
- with tempfile.TemporaryDirectory() as directory:
- path = Path(directory) / "wave.csv"
- path.write_text(content, encoding="utf-8")
- data = load_csv_waveform(path)
-
- self.assertEqual(data.sample_rate, 20_000_000)
- self.assertEqual(data.sample_count, 10)
- self.assertEqual(data.record_count, 3)
- self.assertEqual(data.start_time, 0.0)
- self.assertEqual(data.end_time, 0.2)
- self.assertEqual(data.channels[0].times, [0.0, 0.1])
- self.assertEqual(data.channels[0].levels, [0, 1])
- self.assertEqual(data.channels[1].times, [0.0, 0.2])
- self.assertEqual(data.channels[1].levels, [1, 0])
-
- def test_load_16_channel_bin_waveform(self) -> None:
- """16 通道 BIN 应按小端 16 位采样点读取。"""
- samples = (0x0000, 0x0010, 0x0010, 0x0000)
- with tempfile.TemporaryDirectory() as directory:
- path = Path(directory) / "wave.bin"
- path.write_bytes(struct.pack("<4H", *samples))
- data = load_bin_waveform(path, 10.0, 16, [4])
-
- channel = data.channels[0]
- self.assertEqual(data.sample_count, 4)
- self.assertAlmostEqual(data.duration, 0.3)
- self.assertEqual(channel.index, 4)
- self.assertEqual(channel.times, [0.0, 0.1, 0.3])
- self.assertEqual(channel.levels, [0, 1, 0])
-
- def test_parse_channel_selection(self) -> None:
- """通道输入应支持单个、列表和范围。"""
- self.assertIsNone(parse_channel_selection("自动", 16))
- self.assertEqual(parse_channel_selection("4", 16), [4])
- self.assertEqual(parse_channel_selection("0,2,4-6", 16), [0, 2, 4, 5, 6])
-
- def test_build_frequency_curve_with_automatic_falling_edge(self) -> None:
- """负逻辑脉冲应自动使用第一个真实下降沿计算频率。"""
- channel = WaveformChannel(
- index=4,
- name="Channel 4",
- times=[0.0, 0.1, 0.11, 0.2, 0.21, 0.3, 0.31],
- levels=[1, 0, 1, 0, 1, 0, 1],
- )
- curve = build_frequency_curve(channel, "auto")
-
- self.assertEqual(curve.edge_name, "下降沿")
- self.assertEqual(curve.times, [0.1, 0.2, 0.3])
- for frequency in curve.frequencies:
- self.assertAlmostEqual(frequency, 10.0)
- measured_time, frequency = curve.frequency_at(0.2)
- self.assertEqual(measured_time, 0.2)
- self.assertEqual(frequency, 10.0)
-
- def test_build_frequency_curve_with_rising_edge(self) -> None:
- """指定上升沿时应按相邻上升沿周期换算频率。"""
- channel = WaveformChannel(
- index=0,
- name="Channel 0",
- times=[0.0, 0.01, 0.015, 0.02, 0.0225, 0.025, 0.0275],
- levels=[0, 1, 0, 1, 0, 1, 0],
- )
- curve = build_frequency_curve(channel, "rising")
-
- self.assertEqual(curve.edge_name, "上升沿")
- self.assertAlmostEqual(curve.frequencies[0], 100.0)
- self.assertAlmostEqual(curve.frequencies[-1], 200.0)
-
- def test_parse_marker_times_ms(self) -> None:
- """关键时间应支持多个毫秒数值。"""
- self.assertEqual(parse_marker_times_ms("100, 200，500"), [0.1, 0.2, 0.5])
-
- def test_parse_marker_times_ms_rejects_more_than_ten_unique_values(self) -> None:
- """关键时间超过十个时应给出明确错误。"""
- with self.assertRaisesRegex(CsvParseError, "最多设置 10 个"):
- parse_marker_times_ms(",".join(str(value) for value in range(11)))
-
-
- if __name__ == "__main__":
- unittest.main()
|
