Add selection arrow tests

Author: Gin

SerialPrograms/Source/Tests/PokemonLZA_Tests.cpp

@@ -14,6 +14,7 @@
 #include "NintendoSwitch/NintendoSwitch_ConsoleHandle.h"
 #include "PokemonLZA/Inference/PokemonLZA_DialogDetector.h"
 #include "PokemonLZA/Inference/PokemonLZA_ButtonDetector.h"
+#include "PokemonLZA/Inference/PokemonLZA_SelectionArrowDetector.h"
 #include "PokemonLZA/Inference/PokemonLZA_MainMenuDetector.h"
 #include "PokemonLZA/Inference/Boxes/PokemonLZA_BoxDetection.h"
 #include "PokemonLZA/Inference/Boxes/PokemonLZA_BoxInfoDetector.h" 
@@ -122,6 +123,42 @@ int test_pokemonLZA_MainMenuDetector(const ImageViewRGB32& image, bool target){
     return 0;
 }
 
+int test_pokemonLZA_SelectionArrowDetector(const ImageViewRGB32& image, const std::vector<std::string>& words){
+    // two words: <situation> <True/False>
+    if (words.size() < 2){
+        cerr << "Error: not enough number of words in the filename. Found only " << words.size() << "." << endl;
+        return 1;
+    }
+
+    // Parse situation from second to last word
+    const std::string& situation_str = words[words.size() - 2];
+    SelectionArrowType arrow_type;
+    ImageFloatBox search_box;
+
+    if (situation_str == "Fossil"){
+        arrow_type = SelectionArrowType::RIGHT;
+        search_box = ImageFloatBox(0.6300, 0.4440, 0.2260, 0.3190);
+    }else{
+        cerr << "Error: unknown situation '" << situation_str << "' in filename." << endl;
+        return 1;
+    }
+
+    // Parse True/False from last word
+    bool target = false;
+    if (parse_bool(words[words.size() - 1], target) == false){
+        cerr << "Error: last word in filename should be True or False." << endl;
+        return 1;
+    }
+
+    // Run detector with the specified search box and arrow type
+    auto overlay = DummyVideoOverlay();
+    SelectionArrowDetector detector(COLOR_RED, &overlay, arrow_type, search_box);
+    bool result = detector.detect(image);
+
+    TEST_RESULT_EQUAL(result, target);
+    return 0;
+}
+
 int test_pokemonLZA_BoxCellInfoDetector(const ImageViewRGB32& image, const std::vector<std::string>& words){
     // Expected filename format: <...>_<row>_<col>_<status>.png
     // Where status is one of: Empty, Shiny, Alpha, ShinyAlpha

SerialPrograms/Source/Tests/PokemonLZA_Tests.h

@@ -29,6 +29,8 @@
 
  int test_pokemonLZA_MainMenuDetector(const ImageViewRGB32& image, bool target);
 
+ int test_pokemonLZA_SelectionArrowDetector(const ImageViewRGB32& image, const std::vector<std::string>& words);
+
  int test_pokemonLZA_BoxCellInfoDetector(const ImageViewRGB32& image, const std::vector<std::string>& words);
 
  }

SerialPrograms/Source/Tests/TestMap.cpp

@@ -294,6 +294,7 @@ const std::map<std::string, TestFunction> TEST_MAP = {
     {"PokemonLZA_ButtonDetector", std::bind(image_words_detector_helper, test_pokemonLZA_ButtonDetector, _1)},
     {"PokemonLZA_MainMenuDetector", std::bind(image_bool_detector_helper, test_pokemonLZA_MainMenuDetector, _1)},
     {"PokemonLZA_BoxCellInfoDetector", std::bind(image_words_detector_helper, test_pokemonLZA_BoxCellInfoDetector, _1)},
+    {"PokemonLZA_SelectionArrowDetector", std::bind(image_words_detector_helper, test_pokemonLZA_SelectionArrowDetector, _1)},
 };
 
 TestFunction find_test_function(const std::string& test_space, const std::string& test_name){