util_unittest.py 1.91 KB
Newer Older
1 2 3 4 5
#!/usr/bin/env python
# Copyright 2020 the V8 project authors. All rights reserved.
# Use of this source code is governed by a BSD-style license that can be
# found in the LICENSE file.

6 7 8 9
from __future__ import absolute_import

import os
import sys
10 11
import unittest

12 13 14 15 16 17
TOOLS_PATH = os.path.dirname(os.path.dirname(os.path.dirname(
    os.path.abspath(__file__))))
sys.path.append(TOOLS_PATH)

from testrunner.testproc.util import FixedSizeTopList

18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70
class TestOrderedFixedSizeList(unittest.TestCase):
  def test_empty(self):
    ofsl = FixedSizeTopList(3)
    self.assertEqual(ofsl.as_list(), [])

  def test_12(self):
    ofsl = FixedSizeTopList(3)
    ofsl.add(1)
    ofsl.add(2)
    self.assertEqual(ofsl.as_list(), [2,1])

  def test_4321(self):
    ofsl = FixedSizeTopList(3)
    ofsl.add(4)
    ofsl.add(3)
    ofsl.add(2)
    ofsl.add(1)
    data = ofsl.as_list()
    self.assertEqual(data, [4,3,2])

  def test_544321(self):
    ofsl = FixedSizeTopList(4)
    ofsl.add(5)
    ofsl.add(4)
    ofsl.add(4)
    ofsl.add(3)
    ofsl.add(2)
    ofsl.add(1)
    data = ofsl.as_list()
    self.assertEqual(data, [5, 4, 4, 3])

  def test_withkey(self):
    ofsl = FixedSizeTopList(3,key=lambda x: x['val'])
    ofsl.add({'val':4, 'something': 'four'})
    ofsl.add({'val':3, 'something': 'three'})
    ofsl.add({'val':-1, 'something': 'minusone'})
    ofsl.add({'val':5, 'something': 'five'})
    ofsl.add({'val':0, 'something': 'zero'})
    data = [e['something'] for e in ofsl.as_list()]
    self.assertEqual(data, ['five', 'four', 'three'])

  def test_withkeyclash(self):
    # Test that a key clash does not throw exeption
    ofsl = FixedSizeTopList(2,key=lambda x: x['val'])
    ofsl.add({'val':2, 'something': 'two'})
    ofsl.add({'val':2, 'something': 'two'})
    ofsl.add({'val':0, 'something': 'zero'})
    data = [e['something'] for e in ofsl.as_list()]
    self.assertEqual(data, ['two', 'two'])


if __name__ == '__main__':
  unittest.main()