Skip to content

Commit ed6e7b1

Browse files
aserheichuk-5239chaubold
authored andcommitted
AP-21518: Refactor test in order not to test string representation
1 parent 13fcc7a commit ed6e7b1

File tree

1 file changed

+6
-9
lines changed

1 file changed

+6
-9
lines changed

org.knime.python3.arrow.tests/src/test/python/unittest/test_extension_arrays.py

Lines changed: 6 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -54,7 +54,6 @@
5454
import numpy as np
5555
import pyarrow as pa
5656

57-
import knime._arrow._backend as ka
5857
import knime._arrow._dictencoding as kasde
5958
import knime.api.types as kt
6059
import knime.api.schema as ks
@@ -505,18 +504,16 @@ def test_lru_cache_dict(self):
505504
lru = kasde.LRUCacheDict(cache_len=n)
506505
for i in range(n + 1):
507506
lru[str(i)] = i
508-
correct = "LRUCacheDict([('1', 1), ('2', 2), ('3', 3), ('4', 4), ('5', 5), ('6', 6), ('7', 7), ('8', 8)])"
509-
self.assertEqual(str(lru), correct)
507+
for i in range(1, n):
508+
self.assertEqual(lru[str(i)], i)
510509

511510
# test lru strategy
512-
l = [0, 1, 2, 3, 4]
511+
lst = [0, 1, 2, 3, 4]
513512
lru = kasde.LRUCacheDict(cache_len=n)
514-
for i in l:
513+
for i in lst:
515514
lru[str(i)] = i
516-
self.assertEqual(lru[str(0)], 0)
517-
self.assertEqual(
518-
str(lru), "LRUCacheDict([('1', 1), ('2', 2), ('3', 3), ('4', 4), ('0', 0)])"
519-
)
515+
for i in lst:
516+
self.assertEqual(lru[str(i)], i)
520517

521518

522519
class MyTime:

0 commit comments

Comments
 (0)