fix count with match on arrays

src/Psalm/Internal/Analyzer/Statements/Expression/MatchAnalyzer.php

@@ -79,7 +79,8 @@ public static function analyze(
                 && $stmt->cond->name instanceof PhpParser\Node\Name
                 && ($stmt->cond->name->parts === ['get_class']
                     || $stmt->cond->name->parts === ['gettype']
-                    || $stmt->cond->name->parts === ['get_debug_type'])
+                    || $stmt->cond->name->parts === ['get_debug_type']
+                    || $stmt->cond->name->parts === ['count'])
                 && $stmt->cond->getArgs()
             ) {
                 $first_arg = $stmt->cond->getArgs()[0];

tests/MatchTest.php

@@ -82,6 +82,22 @@ function foo(Foo $value): string {
                 'ignored_issues' => [],
                 'php_version' => '8.0'
             ],
+            'MatchWithCount' => [
+                'code' => '<?php
+                    /**
+                     * @return non-empty-array
+                     */
+                    function test(array $array): array
+                    {
+                        return match (\count($array)) {
+                            0 => throw new \InvalidArgumentException,
+                            default => $array,
+                        };
+                    }',
+                'assertions' => [],
+                'ignored_issues' => [],
+                'php_version' => '8.1'
+            ],
         ];
     }