used assert module. fixed tests