Add tests

Edited by Welser Muñoz