make test work with dict encoded

vaexio · Oct 12, 2021 · 0df4b34 · 0df4b34
1 parent 51b2c06
commit 0df4b34
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/dataframe_protocol_test.py b/tests/dataframe_protocol_test.py
@@ -314,7 +314,7 @@ def assert_buffer_equal(buffer_dtype: Tuple[_VaexBuffer, Any], vaexcol: vaex.exp
     buf, dtype = buffer_dtype
     pytest.raises(NotImplementedError, buf.__dlpack__)
     assert buf.__dlpack_device__() == (1, None)
-    assert dtype[1] == vaexcol.dtype.numpy.itemsize * 8
+    assert dtype[1] == vaexcol.dtype.index_type.numpy.itemsize * 8
     if not isinstance(vaexcol.values, np.ndarray) and isinstance(vaexcol.values.type, pa.DictionaryType):
         assert dtype[2] == vaexcol.index_values().dtype.numpy.str
     else: