kernel/kselftest.py: fix cancellation check for pmu event_code_tests

Move the cancellation check for pmu/event_code_tests to the test
execution block to ensure it correctly verifies the existence of
the required directory and avoids early cancellation.

Signed-off-by: Disha Goel <disgoel@linux.ibm.com>

Author: disgoel

kernel/kselftest.py

@@ -129,8 +129,6 @@ def setUp(self):
                 process.system("make install -C %s" % self.sourcedir,
                                shell=True, sudo=True)
         else:
-            if self.subtest == 'pmu/event_code_tests':
-                self.cancel("selftest not supported on distro")
             # Make sure kernel source repo is configured
             if detected_distro.name in ['centos', 'fedora', 'rhel']:
                 src_name = 'kernel'
@@ -183,6 +181,11 @@ def test(self):
             if self.subtest == "ksm_tests":
                 self.ksmtest()
             else:
+                if self.subtest == 'pmu/event_code_tests':
+                    pmu_test_dir = os.path.join(self.sourcedir,
+                                                'powerpc/pmu/event_code_tests')
+                    if not os.path.exists(pmu_test_dir):
+                        self.cancel("selftest not supported on distro")
                 if self.subtest:
                     test_comp = self.comp + "/" + self.subtest
                 else: