Add unittest asserts