This article introduces a modular simulation platform for assessing thermal energy storage (TES) integrated with air source heat pumps (ASHP). The Python platform is an open-source library that includes classes for modeling air-air and air-water heat pumps, TES devices, and the heating load of residential buildings. To validate the ASHP model, the study utilized experimental data obtained from a commercial heat pump evaluated at the National Renewable Energy Laboratory (NREL). The results indicate a mean deviation of 0.7% for COP across the operating range, with a maximum relative deviation of 12.6%. In terms of system heating capacity, the model had an average deviation of 4.3% compared to experimental results, with a maximum deviation of 8.2%. Three classes were implemented for modeling distinct types of TES devices: a generic TES based on energy balances, a sensible isothermal water tank, and a stratified water tank. Details of the mathematical models are provided, along with their respective strengths and limitations. An example is provided showcasing the integration of a residential 10 kWh thermal storage unit with an ASHP operating in Denver, CO. The comparison of two different discharge criteria for the TES unit highlights the importance of control strategies in the system performance.