Migrate tests to test_that structure