1 files changed, 81 insertions, 0 deletions
diff --git a/host/tests/cal_data_iq_test.cpp b/host/tests/cal_data_iq_test.cpp
new file mode 100644
index 000000000..97b35a331
--- /dev/null
+++ b/host/tests/cal_data_iq_test.cpp
@@ -0,0 +1,81 @@
+//
+// Copyright 2020 Ettus Research, a National Instruments Brand
+//
+// SPDX-License-Identifier: GPL-3.0-or-later
+//
+
+#include <uhd/cal/iq_cal.hpp>
+#include <uhd/exception.hpp>
+#include <boost/test/unit_test.hpp>
+#include <iostream>
+
+using namespace uhd::usrp::cal;
+
+BOOST_AUTO_TEST_CASE(test_iq_cal_api)
+{
+    const std::string name   = "Mock IQ Data";
+    const std::string serial = "ABC1234";
+    const uint64_t timestamp = 0x12340000;
+
+    auto iq_cal_data = iq_cal::make(name, serial, timestamp);
+    BOOST_CHECK_EQUAL(iq_cal_data->get_name(), name);
+    BOOST_CHECK_EQUAL(iq_cal_data->get_serial(), serial);
+    BOOST_CHECK_EQUAL(iq_cal_data->get_timestamp(), timestamp);
+    // Nothing there yet
+    BOOST_REQUIRE_THROW(iq_cal_data->get_cal_coeff(0.0), uhd::assertion_error);
+
+    iq_cal_data->set_cal_coeff(1.0, {1.0, 1.0});
+    iq_cal_data->set_cal_coeff(2.0, {2.0, 2.0});
+
+    // We're in linear interpolation mode
+    auto lo_coeff = iq_cal_data->get_cal_coeff(0.5);
+    auto hi_coeff = iq_cal_data->get_cal_coeff(2.5);
+    BOOST_CHECK_EQUAL(lo_coeff, std::complex<double>(1.0, 1.0));
+    BOOST_CHECK_EQUAL(hi_coeff, std::complex<double>(2.0, 2.0));
+    auto mid_coeff = iq_cal_data->get_cal_coeff(1.75);
+    BOOST_CHECK_EQUAL(mid_coeff, std::complex<double>(1.75, 1.75));
+
+    iq_cal_data->set_interp_mode(interp_mode::NEAREST_NEIGHBOR);
+    lo_coeff = iq_cal_data->get_cal_coeff(0.5);
+    hi_coeff = iq_cal_data->get_cal_coeff(2.5);
+    BOOST_CHECK_EQUAL(lo_coeff, std::complex<double>(1.0, 1.0));
+    BOOST_CHECK_EQUAL(hi_coeff, std::complex<double>(2.0, 2.0));
+    mid_coeff = iq_cal_data->get_cal_coeff(1.75);
+    BOOST_CHECK_EQUAL(mid_coeff, std::complex<double>(2.0, 2.0));
+
+    iq_cal_data->clear();
+    BOOST_REQUIRE_THROW(iq_cal_data->get_cal_coeff(0.0), uhd::assertion_error);
+}
+
+BOOST_AUTO_TEST_CASE(test_iq_cal_serdes)
+{
+    const std::string name   = "Mock IQ Data";
+    const std::string serial = "ABC1234";
+    const uint64_t timestamp = 0x12340000;
+
+    auto iq_cal_data_blueprint = iq_cal::make(name, serial, timestamp);
+    for (double d = 0; d < 5.0; d += 1.0) {
+        iq_cal_data_blueprint->set_cal_coeff(d, {d, d}, d * 10, d * 20);
+    }
+
+    const auto serialized = iq_cal_data_blueprint->serialize();
+
+    auto iq_cal_data = container::make<iq_cal>(serialized);
+
+    BOOST_CHECK_EQUAL(iq_cal_data->get_name(), name);
+    BOOST_CHECK_EQUAL(iq_cal_data->get_serial(), serial);
+    BOOST_CHECK_EQUAL(iq_cal_data->get_timestamp(), timestamp);
+
+    iq_cal_data->set_interp_mode(interp_mode::NEAREST_NEIGHBOR);
+    for (double d = 0; d < 5.0; d += 1.0) {
+        BOOST_CHECK_EQUAL(iq_cal_data->get_cal_coeff(d), std::complex<double>(d, d));
+    }
+}
+
+BOOST_AUTO_TEST_CASE(test_iq_cal_des_fail)
+{
+    std::vector<uint8_t> not_actual_data(42, 23);
+
+    // FIXME: Re-enable this test when the verification works
+    // BOOST_REQUIRE_THROW(container::make<iq_cal>(not_actual_data), uhd::runtime_error);
+}