Finish migrating PLA unit tests to new system.

Author: Mysticial

SerialPrograms/Source/PokemonLA/Inference/Battles/PokemonLA_BattleSpriteWatcher.cpp

@@ -6,11 +6,14 @@
 
 #include <vector>
 #include "Common/Cpp/Logging/AbstractLogger.h"
+#include "Common/Cpp/TestRunners/UnitTestDatabase.h"
+#include "CommonFramework/Globals.h"
 #include "CommonFramework/GlobalSettingsPanel.h"
 #include "CommonFramework/ImageTypes/ImageViewRGB32.h"
 #include "CommonFramework/ImageTools/ImageStats.h"
 #include "CommonFramework/Tools/DebugDumper.h"
 #include "CommonFramework/VideoPipeline/VideoOverlayScopes.h"
+#include "Tests/TestUtils.h"
 #include "PokemonLA_BattleSpriteWatcher.h"
 
 #include <iostream>
@@ -80,6 +83,49 @@ bool BattleSpriteWatcher::set_detected_sprites(const ImageViewRGB32& frame, std:
 }
 
 
+
+
+
+
+
+class Test_BattleSpriteWatcher : public UnitTest{
+public:
+    Test_BattleSpriteWatcher(
+        const std::string& image,
+        bool expected
+    )
+        : UnitTest("PokemonLA::BattleSpriteWatcher - " + image)
+        , m_image(UNIT_TEST_RESOURCE_PATH() + image)
+        , m_expected(expected)
+    {}
+
+    virtual UnitTestResult run(Logger& logger, CancellableScope& scope) const override{
+        DummyVideoOverlay overlay;
+        BattleSpriteWatcher detector(logger, overlay);
+        ImageRGB32 image(m_image);
+        return detector.process_frame(image, current_time()) == m_expected;
+    };
+
+private:
+    std::string m_image;
+    bool m_expected;
+};
+
+
+
+void add_tests_BattleSpriteWatcher(UnitTestDatabase& database){
+
+}
+
+
+
+
+
+
+
+
+
+
 }
 }
 }

SerialPrograms/Source/PokemonLA/Inference/Battles/PokemonLA_BattleSpriteWatcher.h

@@ -62,6 +62,13 @@ class BattleSpriteWatcher : public VisualInferenceCallback{
 };
 
 
+
+
+
+void add_tests_BattleSpriteWatcher(UnitTestDatabase& database);
+
+
+
 }
 }
 }

SerialPrograms/Source/PokemonLA/Inference/Battles/PokemonLA_BattleStartDetector.cpp

@@ -4,11 +4,14 @@
  *
  */
 
+#include "Common/Cpp/TestRunners/UnitTestDatabase.h"
+#include "CommonFramework/Globals.h"
 #include "CommonFramework/ImageTypes/ImageViewRGB32.h"
 #include "CommonFramework/ImageTools/ImageBoxes.h"
 #include "CommonFramework/ImageTools/ImageStats.h"
 #include "CommonFramework/VideoPipeline/VideoOverlayScopes.h"
 #include "CommonTools/Images/ImageGradient.h"
+#include "Tests/TestUtils.h"
 #include "PokemonLA_BattleStartDetector.h"
 
 //#include <iostream>
@@ -62,6 +65,50 @@ bool BattleStartDetector::process_frame(const ImageViewRGB32& frame, WallClock t
 
 
 
+
+
+
+
+
+
+
+
+
+class Test_BattleStartDetector : public UnitTest{
+public:
+    Test_BattleStartDetector(
+        const std::string& image,
+        bool expected
+    )
+        : UnitTest("PokemonLA::BattleStartDetector - " + image)
+        , m_image(UNIT_TEST_RESOURCE_PATH() + image)
+        , m_expected(expected)
+    {}
+
+    virtual UnitTestResult run(Logger& logger, CancellableScope& scope) const override{
+        DummyVideoOverlay overlay;
+        BattleStartDetector detector(logger, overlay);
+        ImageRGB32 image(m_image);
+        return detector.process_frame(image, current_time()) == m_expected;
+    };
+
+private:
+    std::string m_image;
+    bool m_expected;
+};
+
+
+
+
+void add_tests_BattleStartDetector(UnitTestDatabase& database){
+
+}
+
+
+
+
+
+
 }
 }
 }

SerialPrograms/Source/PokemonLA/Inference/Battles/PokemonLA_BattleStartDetector.h

@@ -9,8 +9,8 @@
 #ifndef PokemonAutomation_PokemonLA_BattleStartDetector_H
 #define PokemonAutomation_PokemonLA_BattleStartDetector_H
 
+#include "Common/Cpp/TestRunners/UnitTest.h"
 #include "CommonFramework/ImageTools/ImageBoxes.h"
-#include "CommonFramework/Logging/Logger.h"
 #include "CommonFramework/VideoPipeline/VideoOverlay.h"
 #include "CommonTools/InferenceCallbacks/VisualInferenceCallback.h"
 
@@ -35,6 +35,14 @@ class BattleStartDetector : public VisualInferenceCallback{
 };
 
 
+
+
+
+void add_tests_BattleStartDetector(UnitTestDatabase& database);
+
+
+
+
 }
 }
 }

SerialPrograms/Source/PokemonLA/Inference/Map/PokemonLA_MapMarkerLocator.cpp

@@ -6,12 +6,15 @@
 
 #include <cfloat>
 #include <cmath>
+#include "Common/Cpp/TestRunners/UnitTestDatabase.h"
 #include "Kernels/Waterfill/Kernels_Waterfill_Session.h"
+#include "CommonFramework/Globals.h"
 #include "CommonFramework/ImageTypes/ImageRGB32.h"
 #include "CommonFramework/ImageTools/ImageStats.h"
 #include "CommonFramework/ImageTools/ImageBoxes.h"
 #include "CommonTools/Images/BinaryImage_FilterRgb32.h"
 #include "CommonTools/Images/WaterfillUtilities.h"
+#include "Tests/TestUtils.h"
 #include "PokemonLA_MapMarkerLocator.h"
 
 #include <iostream>
@@ -133,6 +136,48 @@ double get_orientation_on_map(const ImageViewRGB32& screen, bool avoid_lava_area
 
 
 
+
+
+
+
+
+
+
+class Test_MapMarkerLocator : public UnitTest{
+public:
+    Test_MapMarkerLocator(
+        const std::string& image,
+        float target_angle, float threshold
+    )
+        : UnitTest("PokemonLA::MapMarkerLocator - " + image)
+        , m_image(UNIT_TEST_RESOURCE_PATH() + image)
+        , m_target_angle(target_angle)
+        , m_threshold(threshold)
+    {}
+
+    virtual UnitTestResult run(Logger& logger, CancellableScope& scope) const override{
+        ImageRGB32 image(m_image);
+        float angle = get_orientation_on_map(image);
+        TEST_RESULT_APPROXIMATE(angle, m_target_angle, m_threshold);
+        return true;
+    };
+
+private:
+    std::string m_image;
+    float m_target_angle;
+    float m_threshold;
+};
+
+
+
+
+void add_tests_MapMarkerLocator(UnitTestDatabase& database){
+
+}
+
+
+
+
 }
 }
 }

SerialPrograms/Source/PokemonLA/Inference/Map/PokemonLA_MapMarkerLocator.h

@@ -8,6 +8,8 @@
 #ifndef PokemonAutomation_PokemonLA_MapLocation_H
 #define PokemonAutomation_PokemonLA_MapLocation_H
 
+#include "Common/Cpp/TestRunners/UnitTest.h"
+
 namespace PokemonAutomation{
     class ImageViewRGB32;
 namespace NintendoSwitch{
@@ -31,6 +33,11 @@ enum class MapRegion;
 double get_orientation_on_map(const ImageViewRGB32& screen, bool avoid_lava_area = false);
 
 
+
+
+void add_tests_MapMarkerLocator(UnitTestDatabase& database);
+
+
 }
 }
 }

SerialPrograms/Source/PokemonLA/Inference/Map/PokemonLA_MapMissionTabReader.cpp

@@ -4,9 +4,12 @@
  *
  */
 
+#include "Common/Cpp/TestRunners/UnitTestDatabase.h"
+#include "CommonFramework/Globals.h"
 #include "CommonFramework/ImageTools/ImageStats.h"
 #include "CommonFramework/ImageTools/ImageBoxes.h"
 #include "CommonTools/Images/SolidColorTest.h"
+#include "Tests/TestUtils.h"
 #include "PokemonLA_MapMissionTabReader.h"
 
 namespace PokemonAutomation{
@@ -22,6 +25,45 @@ bool is_map_mission_tab_raised(const ImageViewRGB32& screen){
         is_white(image_stats(extract_box_reference(screen, box1)));
 }
 
+
+
+
+
+
+
+
+
+class Test_MapMissionTabReader : public UnitTest{
+public:
+    Test_MapMissionTabReader(
+        const std::string& image,
+        bool expected
+    )
+        : UnitTest("PokemonLA::MapMissionTabReader - " + image)
+        , m_image(UNIT_TEST_RESOURCE_PATH() + image)
+        , m_expected(expected)
+    {}
+
+    virtual UnitTestResult run(Logger& logger, CancellableScope& scope) const override{
+        ImageRGB32 image(m_image);
+        return is_map_mission_tab_raised(image) == m_expected;
+    };
+
+private:
+    std::string m_image;
+    bool m_expected;
+};
+
+
+
+void add_tests_MapMissionTabReader(UnitTestDatabase& database){
+
+}
+
+
+
+
+
 }
 }
 }

SerialPrograms/Source/PokemonLA/Inference/Map/PokemonLA_MapMissionTabReader.h

@@ -8,6 +8,8 @@
 #ifndef PokemonAutomation_PokemonLA_MapMissionTabReader_H
 #define PokemonAutomation_PokemonLA_MapMissionTabReader_H
 
+#include "Common/Cpp/TestRunners/UnitTest.h"
+
 namespace PokemonAutomation{
     class ImageViewRGB32;
 namespace NintendoSwitch{
@@ -20,6 +22,14 @@ namespace PokemonLA{
 bool is_map_mission_tab_raised(const ImageViewRGB32& screen);
 
 
+
+
+
+void add_tests_MapMissionTabReader(UnitTestDatabase& database);
+
+
+
+
 }
 }
 }