Add code for researching an upgrade

python-api-src/lib_unit.cpp

@@ -37,6 +37,7 @@ void define_unit(py::module & m)
         .def("build_target", &Unit::buildTarget)
         .def("train", &Unit::train)
         .def("morph", &Unit::morph)
+        .def("research", &Unit::research)
         .def("__hash__", [](const Unit & unit) { return std::hash<const sc2::Unit *>{}(unit.getUnitPtr()); })
         .def(py::self == py::self)
         .def("__repr__", [](const Unit & unit) { return "<Unit of type: '" + unit.getType().getName() + "'>"; });

src/Unit.cpp

@@ -298,6 +298,12 @@ void Unit::morph(const UnitType & type) const
 #endif
 }
 
+void Unit::research(sc2::UpgradeID upgrade) const
+{
+    BOT_ASSERT(isValid(), "Unit is not valid");
+    m_bot->Actions()->UnitCommand(m_unit, m_bot->Data(upgrade).buildAbility);
+}
+
 bool Unit::isConstructing(const UnitType & type) const
 {
 	sc2::AbilityID buildAbility = m_bot->Data(type).buildAbility;

src/Unit.h

@@ -58,4 +58,5 @@ public:
     void buildTarget    (const UnitType & buildingType, const Unit & target) const;
     void train          (const UnitType & buildingType) const;
     void morph          (const UnitType & type) const;
+    void research       (sc2::UpgradeID upgrade) const;
 };