Files
allwpilib/wpimath/src/test/native/cpp/controller/proto/ArmFeedforwardProtoTest.cpp
PJ Reiniger bb05e20247 [wpimath] Add protobuf/struct for trivial types (#5935)
This implements de/serialization for the types that aren't templated (SwerveDriveKinematics) in C++ or where there is no trivial way to go round-trip (Splines) for the messages.
2023-11-21 10:14:06 -08:00

34 lines
1.1 KiB
C++

// Copyright (c) FIRST and other WPILib contributors.
// Open Source Software; you can modify and/or share it under the terms of
// the WPILib BSD license file in the root directory of this project.
#include <gtest/gtest.h>
#include "controller.pb.h"
#include "frc/controller/ArmFeedforward.h"
using namespace frc;
namespace {
using ProtoType = wpi::Protobuf<frc::ArmFeedforward>;
static constexpr auto Ks = 1.91_V;
static constexpr auto Kg = 2.29_V;
static constexpr auto Kv = 35.04_V * 1_s / 1_rad;
static constexpr auto Ka = 1.74_V * 1_s * 1_s / 1_rad;
const ArmFeedforward kExpectedData{Ks, Kg, Kv, Ka};
} // namespace
TEST(ArmFeedforwardProtoTest, Roundtrip) {
google::protobuf::Arena arena;
google::protobuf::Message* proto = ProtoType::New(&arena);
ProtoType::Pack(proto, kExpectedData);
ArmFeedforward unpacked_data = ProtoType::Unpack(*proto);
EXPECT_EQ(kExpectedData.kS.value(), unpacked_data.kS.value());
EXPECT_EQ(kExpectedData.kG.value(), unpacked_data.kG.value());
EXPECT_EQ(kExpectedData.kV.value(), unpacked_data.kV.value());
EXPECT_EQ(kExpectedData.kA.value(), unpacked_data.kA.value());
}