limboai/tests/test_selector.h

142 lines
4.4 KiB
C
Raw Permalink Normal View History

2023-08-29 14:32:14 +00:00
/**
* test_selector.h
* =============================================================================
2024-03-21 20:38:57 +00:00
* Copyright 2021-2024 Serhii Snitsaruk
2023-08-29 14:32:14 +00:00
*
* Use of this source code is governed by an MIT-style
* license that can be found in the LICENSE file or at
* https://opensource.org/licenses/MIT.
* =============================================================================
*/
#ifndef TEST_SELECTOR_H
#define TEST_SELECTOR_H
2023-08-29 14:56:25 +00:00
#include "limbo_test.h"
2023-08-29 14:32:14 +00:00
#include "modules/limboai/bt/tasks/bt_task.h"
#include "modules/limboai/bt/tasks/composites/bt_selector.h"
namespace TestSelector {
TEST_CASE("[Modules][LimboAI] BTSelector when all return FAILURE") {
Ref<BTSelector> sel = memnew(BTSelector);
Ref<BTTestAction> task1 = memnew(BTTestAction(BTTask::FAILURE));
Ref<BTTestAction> task2 = memnew(BTTestAction(BTTask::FAILURE));
Ref<BTTestAction> task3 = memnew(BTTestAction(BTTask::FAILURE));
sel->add_child(task1);
sel->add_child(task2);
sel->add_child(task3);
REQUIRE(sel->get_child_count() == 3);
// * First execution.
2023-08-30 08:34:12 +00:00
CHECK(sel->execute(0.01666) == BTTask::FAILURE);
2023-08-29 14:32:14 +00:00
CHECK(task1->get_status() == BTTask::FAILURE);
CHECK(task2->get_status() == BTTask::FAILURE);
CHECK(task3->get_status() == BTTask::FAILURE);
2023-08-29 14:56:25 +00:00
CHECK_ENTRIES_TICKS_EXITS(task1, 1, 1, 1);
CHECK_ENTRIES_TICKS_EXITS(task2, 1, 1, 1);
CHECK_ENTRIES_TICKS_EXITS(task3, 1, 1, 1);
2023-08-29 14:32:14 +00:00
// * Second execution.
2023-08-30 08:34:12 +00:00
CHECK(sel->execute(0.01666) == BTTask::FAILURE);
2023-08-29 14:32:14 +00:00
CHECK(task1->get_status() == BTTask::FAILURE);
CHECK(task2->get_status() == BTTask::FAILURE);
CHECK(task3->get_status() == BTTask::FAILURE);
2023-08-29 14:56:25 +00:00
CHECK_ENTRIES_TICKS_EXITS(task1, 2, 2, 2);
CHECK_ENTRIES_TICKS_EXITS(task2, 2, 2, 2);
CHECK_ENTRIES_TICKS_EXITS(task3, 2, 2, 2);
2023-08-29 14:32:14 +00:00
}
TEST_CASE("[Modules][LimboAI] BTSelector when second returns SUCCESS") {
Ref<BTSelector> sel = memnew(BTSelector);
Ref<BTTestAction> task1 = memnew(BTTestAction(BTTask::FAILURE));
Ref<BTTestAction> task2 = memnew(BTTestAction(BTTask::SUCCESS));
Ref<BTTestAction> task3 = memnew(BTTestAction(BTTask::FAILURE));
sel->add_child(task1);
sel->add_child(task2);
sel->add_child(task3);
REQUIRE(sel->get_child_count() == 3);
// * First execution.
2023-08-30 08:34:12 +00:00
CHECK(sel->execute(0.01666) == BTTask::SUCCESS);
2023-08-29 14:32:14 +00:00
CHECK(task1->get_status() == BTTask::FAILURE);
CHECK(task2->get_status() == BTTask::SUCCESS);
CHECK(task3->get_status() == BTTask::FRESH);
2023-08-29 14:56:25 +00:00
CHECK_ENTRIES_TICKS_EXITS(task1, 1, 1, 1);
CHECK_ENTRIES_TICKS_EXITS(task2, 1, 1, 1);
CHECK_ENTRIES_TICKS_EXITS(task3, 0, 0, 0);
2023-08-29 14:32:14 +00:00
// * Second execution.
2023-08-30 08:34:12 +00:00
CHECK(sel->execute(0.01666) == BTTask::SUCCESS);
2023-08-29 14:32:14 +00:00
CHECK(task1->get_status() == BTTask::FAILURE);
CHECK(task2->get_status() == BTTask::SUCCESS);
CHECK(task3->get_status() == BTTask::FRESH);
2023-08-29 14:56:25 +00:00
CHECK_ENTRIES_TICKS_EXITS(task1, 2, 2, 2);
CHECK_ENTRIES_TICKS_EXITS(task2, 2, 2, 2);
CHECK_ENTRIES_TICKS_EXITS(task3, 0, 0, 0);
2023-08-29 14:32:14 +00:00
}
TEST_CASE("[Modules][LimboAI] BTSelector when second returns RUNNING") {
Ref<BTSelector> sel = memnew(BTSelector);
Ref<BTTestAction> task1 = memnew(BTTestAction(BTTask::FAILURE));
Ref<BTTestAction> task2 = memnew(BTTestAction(BTTask::RUNNING));
Ref<BTTestAction> task3 = memnew(BTTestAction(BTTask::FAILURE));
sel->add_child(task1);
sel->add_child(task2);
sel->add_child(task3);
REQUIRE(sel->get_child_count() == 3);
// * First execution.
2023-08-30 08:34:12 +00:00
CHECK(sel->execute(0.01666) == BTTask::RUNNING);
2023-08-29 14:32:14 +00:00
CHECK(task1->get_status() == BTTask::FAILURE);
CHECK(task2->get_status() == BTTask::RUNNING);
CHECK(task3->get_status() == BTTask::FRESH);
2023-08-29 14:56:25 +00:00
CHECK_ENTRIES_TICKS_EXITS(task1, 1, 1, 1);
CHECK_ENTRIES_TICKS_EXITS(task2, 1, 1, 0);
CHECK_ENTRIES_TICKS_EXITS(task3, 0, 0, 0);
2023-08-29 14:32:14 +00:00
// * Second execution.
2023-08-30 08:34:12 +00:00
CHECK(sel->execute(0.01666) == BTTask::RUNNING);
2023-08-29 14:32:14 +00:00
CHECK(task1->get_status() == BTTask::FAILURE);
CHECK(task2->get_status() == BTTask::RUNNING);
CHECK(task3->get_status() == BTTask::FRESH);
2023-08-29 14:56:25 +00:00
CHECK_ENTRIES_TICKS_EXITS(task1, 1, 1, 1);
CHECK_ENTRIES_TICKS_EXITS(task2, 1, 2, 0);
CHECK_ENTRIES_TICKS_EXITS(task3, 0, 0, 0);
2023-08-29 14:32:14 +00:00
// * Third execution with second task returning FAILURE.
task2->ret_status = BTTask::FAILURE;
2023-08-30 08:34:12 +00:00
CHECK(sel->execute(0.01666) == BTTask::FAILURE);
2023-08-29 14:32:14 +00:00
CHECK(task1->get_status() == BTTask::FAILURE);
CHECK(task2->get_status() == BTTask::FAILURE);
CHECK(task3->get_status() == BTTask::FAILURE);
2023-08-29 14:56:25 +00:00
CHECK_ENTRIES_TICKS_EXITS(task1, 1, 1, 1);
CHECK_ENTRIES_TICKS_EXITS(task2, 1, 3, 1);
CHECK_ENTRIES_TICKS_EXITS(task3, 1, 1, 1);
2023-08-29 14:32:14 +00:00
}
} //namespace TestSelector
#endif // TEST_SELECTOR_H