change assert to test