如何对在AndroidViewModel类中进行改进调用的方法进行单元测试？

import androidx.lifecycle.ViewModelProvider;
import androidx.test.ext.junit.runners.AndroidJUnit4;
import androidx.test.platform.app.InstrumentationRegistry;

import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;

import static org.junit.Assert.assertEquals;

@RunWith(AndroidJUnit4.class)
public class YourViewModelTest {

    private YourViewModel yourViewModel;

    @Before
    public void setup() {
        // 在测试方法执行之前，进行初始化操作
        yourViewModel = new ViewModelProvider.AndroidViewModelFactory(
                InstrumentationRegistry.getInstrumentation().getTargetContext().getApplication())
                .create(YourViewModel.class);
    }

    @Test
    public void testYourMethod() {
        // 调用你要测试的方法，并获取返回结果
        String result = yourViewModel.yourMethod();

        // 使用断言来验证返回结果是否符合预期
        assertEquals("Expected result", result);
    }
}