Update ConstantArrayType.php

Author: staabm

src/Type/Constant/ConstantArrayType.php

@@ -702,7 +702,12 @@ public function setOffsetValueType(?Type $offsetType, Type $valueType, bool $uni
 	{
 		if ($offsetType !== null) {
 			$scalarKeyTypes = $this->resolveFiniteScalarKeyTypes($offsetType);
-			if ($scalarKeyTypes !== null && count($scalarKeyTypes) >= 2) {
+			// turn into tagged union for more precise results
+			if (
+				$scalarKeyTypes !== null
+				&& count($scalarKeyTypes) >= 2
+				&& count($scalarKeyTypes) <= self::CHUNK_FINITE_TYPES_LIMIT
+			) {
 				$hasNewKey = false;
 				foreach ($scalarKeyTypes as $scalarKeyType) {
 					$existingKeyFound = false;
@@ -770,11 +775,7 @@ private function resolveFiniteScalarKeyTypes(Type $offsetType): ?array
 			}
 		}
 
-		if (count($result) <= self::CHUNK_FINITE_TYPES_LIMIT) {
-			return $result;
-		}
-
-		return null;
+		return $result;
 	}
 
 	public function unsetOffset(Type $offsetType): Type