How to run android instrumental test using `androidx` and `navgation architecture`?

import  android.support.v7.widget.Toolbar;

import android.widget.Toolbar;

@RunWith(Suite.class)
@Suite.SuiteClasses({TestEnvironmentInitializer.class,
        FirstTest.class,
        SecondTest.class
})
public class MyTestSuite {
}

 ./gradlew yourappmodule:connectedAndroidTest -Pandroid.testInstrumentationRunnerArguments.class=your.package.androidtest.MyTestSuite 

./gradlew yourappmodule:connectedStagingAndroidTest -Pandroid.testInstrumentationRunnerArguments.class=your.package.androidtest.MyTestSuite 

@Rule
public GrantPermissionRule runtimePermissionRule = GrantPermissionRule.grant(
        Manifest.permission.READ_EXTERNAL_STORAGE, Manifest.permission.WRITE_EXTERNAL_STORAGE);

@RunWith(AndroidJUnit4.class)
@LargeTest
public class TestEnvironmentInitializer {

    @Rule
    public GrantPermissionRule runtimePermissionRule = GrantPermissionRule.grant(
            Manifest.permission.READ_EXTERNAL_STORAGE, Manifest.permission.WRITE_EXTERNAL_STORAGE);


    @Test
    public void initEnvironment() {
        try {

            Context testContext = InstrumentationRegistry.getInstrumentation().getContext();


            File root = Environment.getExternalStorageDirectory();


            File testDataFolder = new File(root, "testData");

            if(testDataFolder.mkdirs()) {

                File destFile = new File(testDataFolder, "test_data.dat");
                if (!destFile.exists()) {
                    copyAssetToExternalStorage(testContext, "assetssubfolder/test_data.dat", destFile);
                }
            }

        } catch (Throwable e) {
            System.exit(-1);
        }
    }

    void copyAssetToExternalStorage(Context context, String fileToCopy, File dest) throws
            IOException {

        try (InputStream src = context.getAssets().open(fileToCopy)) {

            try (FileOutputStream destStream = new FileOutputStream(
                    dest)) {

                copyStream(src, destStream);
            }
        }
    }

    void copyStream(InputStream src, OutputStream dest) throws IOException {

        byte[] buffer = new byte[1024];
        int read;
        while ((read = src.read(buffer)) != -1) {
            dest.write(buffer, 0, read);
        }
    }
}

Log.d(TAG, message)

androidTestImplementation 'androidx.test:rules:1.2.0'

public class DataActivity extends AppCompatActivity {

    private DataViewModel mViewModel;
    private DataActivityBinding mBinding;

    @Override
    protected void onCreate(Bundle savedInstanceState) {
        super.onCreate(savedInstanceState);

        this.mViewModel = new DataViewModel();
        this.mBinding = DataBindingUtil.setContentView(this, R.layout.data_activity);
        this.mBinding.setViewModel(this.mViewModel);
    }
}