Add tests

[fristonio]

Old tests are after the refactor we will need to add tests for origami-lib.

[gupta-utkarsh]

Can you list down the modules in this issue for which the tests are required? Create a sub-issue for each module and list the features for which tests need to be written. This would help in tracking the progress of tests for origami-lib.