Add PNP and Multitag packet tests

PhotonVision · Nov 19, 2023 · 1e95d37 · 1e95d37
1 parent eca609c
commit 1e95d37
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/photon-targeting/src/test/native/cpp/PacketTest.cpp b/photon-targeting/src/test/native/cpp/PacketTest.cpp
@@ -21,9 +21,33 @@
 
 #include "gtest/gtest.h"
 #include "photon/dataflow/structures/Packet.h"
+#include "photon/targeting/MultiTargetPNPResult.h"
+#include "photon/targeting/PNPResult.h"
 #include "photon/targeting/PhotonPipelineResult.h"
 #include "photon/targeting/PhotonTrackedTarget.h"
 
+TEST(PacketTest, PNPResult) {
+  photon::PNPResult result;
+  photon::Packet p;
+  p << result;
+
+  photon::PNPResult b;
+  p >> b;
+
+  EXPECT_EQ(result, b);
+}
+
+TEST(PacketTest, MultiTargetPNPResult) {
+  photon::MultiTargetPNPResult result;
+  photon::Packet p;
+  p << result;
+
+  photon::MultiTargetPNPResult b;
+  p >> b;
+
+  EXPECT_EQ(result, b);
+}
+
 TEST(PacketTest, PhotonTrackedTarget) {
   photon::PhotonTrackedTarget target{
       3.0,