Add tests

[victor-villar]

We need to add tests to the repo.
Ideally:

• Rust unit tests
• Python unit tests
• Integration/full execution tests

[cbjuan]

is this done?