Add basic testing for new widgets