Pārlūkot izejas kodu

refactor: move some tests from test_config.py to config/test_0x08_pktctrl0.py

Fabian Peter Hammerle 3 gadi atpakaļ
vecāks
revīzija
280da5968c
2 mainītis faili ar 92 papildinājumiem un 69 dzēšanām
  1. 92 0
      tests/config/test_0x08_pktctrl0.py
  2. 0 69
      tests/test_config.py

+ 92 - 0
tests/config/test_0x08_pktctrl0.py

@@ -0,0 +1,92 @@
+# python-cc1101 - Python Library to Transmit RF Signals via C1101 Transceivers
+#
+# Copyright (C) 2020 Fabian Peter Hammerle <fabian@hammerle.me>
+#
+# This program is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# any later version.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with this program.  If not, see <https://www.gnu.org/licenses/>.
+
+import unittest.mock
+
+import pytest
+
+from cc1101.options import PacketLengthMode
+
+# pylint: disable=protected-access
+
+
+@pytest.mark.parametrize(
+    ("pktctrl0_before", "pktctrl0_after"),
+    (
+        # unchanged
+        (0b00000000, 0b00000000),
+        (0b00010000, 0b00010000),
+        (0b00010001, 0b00010001),
+        (0b01000000, 0b01000000),
+        (0b01000010, 0b01000010),
+        (0b01110000, 0b01110000),
+        (0b01110010, 0b01110010),
+        # disabled
+        (0b00010100, 0b00010000),
+        (0b01000100, 0b01000000),
+        (0b01000110, 0b01000010),
+        (0b01110110, 0b01110010),
+    ),
+)
+def test_disable_checksum(transceiver, pktctrl0_before, pktctrl0_after):
+    xfer_mock = transceiver._spi.xfer
+    xfer_mock.return_value = [15, 15]
+    with unittest.mock.patch.object(
+        transceiver, "_read_single_byte", return_value=pktctrl0_before
+    ):
+        transceiver.disable_checksum()
+    xfer_mock.assert_called_once_with([0x08 | 0x40, pktctrl0_after])
+
+
+@pytest.mark.parametrize(
+    ("pktctrl0", "expected_mode"),
+    (
+        (0b00000000, PacketLengthMode.FIXED),
+        (0b00000001, PacketLengthMode.VARIABLE),
+        (0b01000100, PacketLengthMode.FIXED),
+        (0b01000101, PacketLengthMode.VARIABLE),
+    ),
+)
+def test_get_packet_length_mode(transceiver, pktctrl0, expected_mode):
+    xfer_mock = transceiver._spi.xfer
+    xfer_mock.return_value = [0, pktctrl0]
+    assert transceiver.get_packet_length_mode() == expected_mode
+    xfer_mock.assert_called_once_with([0x08 | 0x80, 0])
+
+
+@pytest.mark.parametrize(
+    ("pktctrl0_before", "pktctrl0_after", "mode"),
+    (
+        (0b00000000, 0b00000000, PacketLengthMode.FIXED),
+        (0b00000001, 0b00000000, PacketLengthMode.FIXED),
+        (0b00000001, 0b00000001, PacketLengthMode.VARIABLE),
+        (0b00000010, 0b00000000, PacketLengthMode.FIXED),
+        (0b00000010, 0b00000001, PacketLengthMode.VARIABLE),
+        (0b01000100, 0b01000100, PacketLengthMode.FIXED),
+        (0b01000100, 0b01000101, PacketLengthMode.VARIABLE),
+        (0b01000101, 0b01000100, PacketLengthMode.FIXED),
+        (0b01000101, 0b01000101, PacketLengthMode.VARIABLE),
+    ),
+)
+def test_set_packet_length_mode(transceiver, pktctrl0_before, pktctrl0_after, mode):
+    xfer_mock = transceiver._spi.xfer
+    xfer_mock.return_value = [15, 15]
+    with unittest.mock.patch.object(
+        transceiver, "_read_single_byte", return_value=pktctrl0_before
+    ):
+        transceiver.set_packet_length_mode(mode)
+    xfer_mock.assert_called_once_with([0x08 | 0x40, pktctrl0_after])

+ 0 - 69
tests/test_config.py

@@ -21,7 +21,6 @@ import warnings
 import pytest
 
 import cc1101
-from cc1101.options import PacketLengthMode
 
 # pylint: disable=protected-access
 
@@ -259,74 +258,6 @@ def test_set_packet_length_bytes_fail(transceiver, packet_length):
     transceiver._spi.xfer.assert_not_called()
 
 
-@pytest.mark.parametrize(
-    ("pktctrl0_before", "pktctrl0_after"),
-    (
-        # unchanged
-        (0b00000000, 0b00000000),
-        (0b00010000, 0b00010000),
-        (0b00010001, 0b00010001),
-        (0b01000000, 0b01000000),
-        (0b01000010, 0b01000010),
-        (0b01110000, 0b01110000),
-        (0b01110010, 0b01110010),
-        # disabled
-        (0b00010100, 0b00010000),
-        (0b01000100, 0b01000000),
-        (0b01000110, 0b01000010),
-        (0b01110110, 0b01110010),
-    ),
-)
-def test_disable_checksum(transceiver, pktctrl0_before, pktctrl0_after):
-    xfer_mock = transceiver._spi.xfer
-    xfer_mock.return_value = [15, 15]
-    with unittest.mock.patch.object(
-        transceiver, "_read_single_byte", return_value=pktctrl0_before
-    ):
-        transceiver.disable_checksum()
-    xfer_mock.assert_called_once_with([0x08 | 0x40, pktctrl0_after])
-
-
-@pytest.mark.parametrize(
-    ("pktctrl0", "expected_mode"),
-    (
-        (0b00000000, PacketLengthMode.FIXED),
-        (0b00000001, PacketLengthMode.VARIABLE),
-        (0b01000100, PacketLengthMode.FIXED),
-        (0b01000101, PacketLengthMode.VARIABLE),
-    ),
-)
-def test_get_packet_length_mode(transceiver, pktctrl0, expected_mode):
-    xfer_mock = transceiver._spi.xfer
-    xfer_mock.return_value = [0, pktctrl0]
-    assert transceiver.get_packet_length_mode() == expected_mode
-    xfer_mock.assert_called_once_with([0x08 | 0x80, 0])
-
-
-@pytest.mark.parametrize(
-    ("pktctrl0_before", "pktctrl0_after", "mode"),
-    (
-        (0b00000000, 0b00000000, PacketLengthMode.FIXED),
-        (0b00000001, 0b00000000, PacketLengthMode.FIXED),
-        (0b00000001, 0b00000001, PacketLengthMode.VARIABLE),
-        (0b00000010, 0b00000000, PacketLengthMode.FIXED),
-        (0b00000010, 0b00000001, PacketLengthMode.VARIABLE),
-        (0b01000100, 0b01000100, PacketLengthMode.FIXED),
-        (0b01000100, 0b01000101, PacketLengthMode.VARIABLE),
-        (0b01000101, 0b01000100, PacketLengthMode.FIXED),
-        (0b01000101, 0b01000101, PacketLengthMode.VARIABLE),
-    ),
-)
-def test_set_packet_length_mode(transceiver, pktctrl0_before, pktctrl0_after, mode):
-    xfer_mock = transceiver._spi.xfer
-    xfer_mock.return_value = [15, 15]
-    with unittest.mock.patch.object(
-        transceiver, "_read_single_byte", return_value=pktctrl0_before
-    ):
-        transceiver.set_packet_length_mode(mode)
-    xfer_mock.assert_called_once_with([0x08 | 0x40, pktctrl0_after])
-
-
 @pytest.mark.parametrize(
     ("freq_hz", "warn"),
     (