Task/add unit tests

#### Reviewers
@jframos @flopezag @jesuspg 

#### Description
ADD unit tests

#### Test
tested locally