AutomatedTestCase/main.py at main · Bia104/AutomatedTestCase · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
from Models.agent_learning import AgentLearning
from Models.environment_simulation import TaxiGymEnv
from Models.prioritization import Prioritizer
from Models.test_case_generator import TestCaseGenerator

MODEL_PATH = "ppo_elements/best_model1.zip"
TEST_CASES_PATH = "test_cases/generated_test_cases.json"


def generate_test_cases(model, env, count=100):
    print("Generating test cases...")
    generator = TestCaseGenerator(model, env)
    generator.generate(count)
    print(f"Saved {count} test cases.")


def prioritize_test_cases(test_cases):
    print("Prioritizing test cases...")
    prioritizer = Prioritizer(test_cases)
    prioritized = prioritizer.prioritize("./test_cases/")
    print("Prioritization complete.")
    return prioritized


def main():

    env = TaxiGymEnv()

    # Train or load the agent
    learning = AgentLearning(env)
    learning.load_or_train_model()

    # Generate test cases
    generate_test_cases(MODEL_PATH, env)

    # Prioritize test cases
    prioritized_cases = prioritize_test_cases(TEST_CASES_PATH)
    prioritized_cases = [item[1] for item in prioritized_cases]
    print("Prioritized test cases:", prioritized_cases)


if __name__ == "__main__":
    main()