From 98fdcf130f2e1b3bd7a0ae85f350bb7f07d4da33 Mon Sep 17 00:00:00 2001
From: Daniele Forsi IU5HKX <iu5hkx@gmail.com>
Date: Sun, 4 May 2025 21:42:10 +0200
Subject: [PATCH] Add tests for the Amp object

---
 bindings/python/test_amp.py | 78 +++++++++++++++++++++++++++++++++++++
 1 file changed, 78 insertions(+)
 create mode 100755 bindings/python/test_amp.py

diff --git a/bindings/python/test_amp.py b/bindings/python/test_amp.py
new file mode 100755
index 000000000..0673d644c
--- /dev/null
+++ b/bindings/python/test_amp.py
@@ -0,0 +1,78 @@
+#! /bin/env pytest
+"""Tests of the Python bindings for Hamlib
+
+Running this script directly will use the installed bindings.
+For an in-tree run use "make check", or set PYTHONPATH to point to
+the directories containing Hamlib.py and _Hamlib.so.
+"""
+import Hamlib
+
+Hamlib.rig_set_debug(Hamlib.RIG_DEBUG_NONE)
+
+AMP_MODEL = Hamlib.AMP_MODEL_DUMMY
+
+class TestClass:
+    """Container class for tests"""
+
+    def test_without_open(self):
+        """Call all the methods that do not depend on open()"""
+        amp = Hamlib.Amp(AMP_MODEL)
+        assert amp is not None
+        assert amp.do_exception == 0
+        assert amp.error_status == Hamlib.RIG_OK
+
+        assert isinstance(amp.caps.model_name, str)
+        assert isinstance(amp.caps.mfg_name, str)
+        assert isinstance(amp.caps.version, str)
+        assert isinstance(amp.caps.copyright, str)
+
+        assert amp.set_conf("", "") is None
+        assert amp.get_conf("") == ""
+        assert amp.get_conf(0) == ""
+        conf = amp.get_conf("mcfg")
+        assert isinstance(conf, str)
+        assert amp.set_conf("mcfg", "foo") is None
+        conf = amp.get_conf("mcfg")
+        assert conf == ""  # FIXME: should return "foo"
+
+        assert amp.token_lookup("") is None
+
+
+    def test_with_open(self):
+        """Call all the methods that depend on open()"""
+        amp = Hamlib.Amp(AMP_MODEL)
+        assert amp is not None
+
+        assert amp.state.comm_state == 0
+        assert amp.open() is None
+        assert amp.state.comm_state == 1
+        info = amp.get_info()
+        assert isinstance(info, str)
+        assert amp.reset(Hamlib.AMP_RESET_FAULT) is None
+        assert amp.set_freq(0) is None
+        assert amp.set_freq(123.45) is None
+        assert amp.get_freq() == 123.45
+        assert amp.get_level(Hamlib.AMP_LEVEL_NONE) is None
+        level = amp.get_level(Hamlib.AMP_LEVEL_SWR)
+        assert isinstance(level, float)
+        level = amp.get_level(Hamlib.AMP_LEVEL_PWR_REFLECTED)
+        assert isinstance(level, int)
+        level = amp.get_level(Hamlib.AMP_LEVEL_FAULT)
+        assert isinstance(level, str)
+        level = amp.get_level(123456)
+        assert level is None
+        assert amp.set_powerstat(Hamlib.RIG_POWER_ON) is None
+        assert amp.get_powerstat() == Hamlib.RIG_POWER_ON
+        assert amp.close() is None
+        assert amp.state.comm_state == 0
+        info = amp.get_info()
+        assert info is None
+
+
+    def test_object_creation(self):
+        """Create all objects available"""
+        amp = Hamlib.Rig(AMP_MODEL)
+        assert amp is not None
+
+        assert isinstance(amp.caps, Hamlib.rig_caps)
+        assert isinstance(amp.state, Hamlib.rig_state)