Closes sebastianbergmann/phpunit#2410

Author: sebastianbergmann

src/CodeCoverage.php

@@ -10,6 +10,8 @@
 
 namespace SebastianBergmann\CodeCoverage;
 
+use PHPUnit\Framework\TestCase;
+use PHPUnit\Runner\PhptTestCase;
 use SebastianBergmann\CodeCoverage\Driver\Driver;
 use SebastianBergmann\CodeCoverage\Driver\Xdebug;
 use SebastianBergmann\CodeCoverage\Driver\HHVM;
@@ -340,7 +342,7 @@ public function append(array $data, $id = null, $append = true, $linesToBeCovere
         $size   = 'unknown';
         $status = null;
 
-        if ($id instanceof \PHPUnit_Framework_TestCase) {
+        if ($id instanceof TestCase) {
             $_size = $id->getSize();
 
             if ($_size == \PHPUnit\Util\Test::SMALL) {
@@ -353,7 +355,7 @@ public function append(array $data, $id = null, $append = true, $linesToBeCovere
 
             $status = $id->getStatus();
             $id     = get_class($id) . '::' . $id->getName();
-        } elseif ($id instanceof \PHPUnit_Extensions_PhptTestCase) {
+        } elseif ($id instanceof PhptTestCase) {
             $size = 'large';
             $id   = $id->getName();
         }
@@ -619,7 +621,7 @@ private function applyCoversAnnotationFilter(array &$data, $linesToBeCovered, ar
         }
 
         if ($this->checkForUnintentionallyCoveredCode &&
-            (!$this->currentId instanceof \PHPUnit_Framework_TestCase ||
+            (!$this->currentId instanceof TestCase ||
             (!$this->currentId->isMedium() && !$this->currentId->isLarge()))) {
             $this->performUnintentionallyCoveredCodeCheck(
                 $data,