diff --git a/tests/include/aocTests/TestContext.hpp b/tests/include/aocTests/TestContext.hpp
index 9467006..dcdebfd 100644
--- a/tests/include/aocTests/TestContext.hpp
+++ b/tests/include/aocTests/TestContext.hpp
@@ -26,6 +26,8 @@ class TestContext
 public:
     void run(const std::unique_ptr<Solver>&& solver, const long long int expected1, const long long int expected2,
         const std::vector<std::string>& inputPaths);
+    void runPart2(const std::unique_ptr<Solver>&& solver, const long long int expected,
+        const std::vector<std::string>& inputPaths);
     std::vector<std::string> getInputPaths() const;
     std::vector<std::string> getExampleInputPaths() const;
 };
diff --git a/tests/src/TestContext.cpp b/tests/src/TestContext.cpp
index 9f71caf..f3697be 100644
--- a/tests/src/TestContext.cpp
+++ b/tests/src/TestContext.cpp
@@ -29,6 +29,15 @@ void TestContext::run(const std::unique_ptr<Solver>&& solver, const long long in
     REQUIRE(expected2 == solver->getResultPart2());
 }
 
+void TestContext::runPart2(const std::unique_ptr<Solver>&& solver, const long long int expected,
+    const std::vector<std::string>& inputPaths)
+{
+    SolverEngine solverEngine{ inputPaths };
+    solverEngine.run(*solver);
+
+    REQUIRE(expected == solver->getResultPart2());
+}
+
 std::vector<std::string> TestContext::getInputPaths() const
 {
     return std::vector<std::string>{ "data", "../../../data", "../../../../data" };