Feature update unittests

Update old unit tests so that they check modified methods in both library and CLI

@iliatsi or @cstavr may review and merge it