move tests to sample_problems.py