Create first PyTest of migration code

[HenrikSommer]

Tasks:

• [ ] Extract code from notebook to a function in a separate .py file
• [ ] Create first PyTest of the extracted function

[HenrikSommer]

Wrong repo