combined unit tests into one class