Add missing tests for internal heap index

Change-Id: If3705ef86c54504c930888829f6e38f88cdbd1ef
Signed-off-by: Jablonski, Mateusz <[email protected]>

Author: JablonskiMateusz

unit_tests/os_interface/windows/wddm20_tests.cpp

@@ -1042,3 +1042,15 @@ TEST_F(WddmHeapSelectorTest, givenFullAddressSpaceWhenSelectingHeapForExternalAl
     }
     EXPECT_EQ(HeapIndex::HEAP_STANDARD, wddm->selectHeap(&allocation, nullptr));
 }
+
+TEST(WddmInternalHeapTest, whenConfigurationIs64BitThenInternalHeapIndexIsHeapInternalDeviceMemory) {
+    if (is64bit) {
+        EXPECT_EQ(HeapIndex::HEAP_INTERNAL_DEVICE_MEMORY, internalHeapIndex);
+    }
+}
+
+TEST(WddmInternalHeapTest, whenConfigurationIs32BitThenInternalHeapIndexIsHeapInternal) {
+    if (is32bit) {
+        EXPECT_EQ(HeapIndex::HEAP_INTERNAL, internalHeapIndex);
+    }
+}