Make sure that output dictionary is same as input dictionary

trinodb · Oct 11, 2021 · be4ce14 · be4ce14
1 parent 73ad472
commit be4ce14
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/...trino-main/src/test/java/io/trino/operator/project/TestDictionaryAwarePageProjection.java b/...trino-main/src/test/java/io/trino/operator/project/TestDictionaryAwarePageProjection.java
@@ -197,7 +197,10 @@ public void testPreservesDictionaryInstance()
         assertInstanceOf(secondOutputBlock, DictionaryBlock.class);
 
         assertNotSame(firstOutputBlock, secondOutputBlock);
-        assertSame(((DictionaryBlock) firstOutputBlock).getDictionary(), ((DictionaryBlock) secondOutputBlock).getDictionary());
+        Block firstDictionary = ((DictionaryBlock) firstOutputBlock).getDictionary();
+        Block secondDictionary = ((DictionaryBlock) secondOutputBlock).getDictionary();
+        assertSame(firstDictionary, secondDictionary);
+        assertSame(firstDictionary, dictionary);
     }
 
     private static DictionaryBlock createDictionaryBlock(int dictionarySize, int blockSize)