put in sample tests from docs