add unit test

#### Reviewers
@jframos @flopezag 

#### Description
 ADD unit tests

#### Test
Tested locally