summary refs log tree commit diff stats
path: root/tests/compiler/tbrees.nim
blob: 5f6482ed94ecc13e466f31a55d5831e92c41f5f0 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
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
71
72
73
74
75
76
77
78
79
80
81
82
83
84
discard """
  output: '''
www.amazon.com: 207.171.182.16
www.apple.com: 17.112.152.32
www.cnn.com: 64.236.16.20
www.cs.princeton.edu: abc
www.dell.com: 143.166.224.230
www.ebay.com: 66.135.192.87
www.espn.com: 199.181.135.201
www.google.com: 216.239.41.99
www.microsoft.com: 207.126.99.140
www.nytimes.com: 199.239.136.200
www.princeton.edu: 128.112.128.15
www.simpsons.com: 209.052.165.60
www.slashdot.org: 66.35.250.151
www.weather.com: 63.111.66.11
www.yahoo.com: 216.109.118.65
www.yale.edu: 130.132.143.21
'''
"""

import compiler/btrees

import random, tables

proc main =
  var st = initBTree[string, string]()
  st.add("www.cs.princeton.edu", "abc")
  st.add("www.princeton.edu",    "128.112.128.15")
  st.add("www.yale.edu",         "130.132.143.21")
  st.add("www.simpsons.com",     "209.052.165.60")
  st.add("www.apple.com",        "17.112.152.32")
  st.add("www.amazon.com",       "207.171.182.16")
  st.add("www.ebay.com",         "66.135.192.87")
  st.add("www.cnn.com",          "64.236.16.20")
  st.add("www.google.com",       "216.239.41.99")
  st.add("www.nytimes.com",      "199.239.136.200")
  st.add("www.microsoft.com",    "207.126.99.140")
  st.add("www.dell.com",         "143.166.224.230")
  st.add("www.slashdot.org",     "66.35.250.151")
  st.add("www.espn.com",         "199.181.135.201")
  st.add("www.weather.com",      "63.111.66.11")
  st.add("www.yahoo.com",        "216.109.118.65")

  doAssert st.getOrDefault("www.cs.princeton.edu") == "abc"
  doAssert st.getOrDefault("www.harvardsucks.com") == ""

  doAssert st.getOrDefault("www.simpsons.com") == "209.052.165.60"
  doAssert st.getOrDefault("www.apple.com") == "17.112.152.32"
  doAssert st.getOrDefault("www.ebay.com") == "66.135.192.87"
  doAssert st.getOrDefault("www.dell.com") == "143.166.224.230"
  doAssert(st.len == 16)

  for k, v in st:
    echo k, ": ", v

  when false:
    var b2 = initBTree[string, string]()
    const iters = 10_000
    for i in 1..iters:
      b2.add($i, $(iters - i))
    for i in 1..iters:
      let x = b2.getOrDefault($i)
      if x != $(iters - i):
        echo "got ", x, ", but expected ", iters - i
    echo b2.len

  when true:
    var b2 = initBTree[int, string]()
    var t2 = initTable[int, string]()
    const iters = 100_000
    for i in 1..iters:
      let x = rand(high(int))
      if not t2.hasKey(x):
        doAssert b2.getOrDefault(x).len == 0, " what, tree has this element " & $x
        t2[x] = $x
        b2.add(x, $x)

    doAssert(b2.len == t2.len, "unique entries " & $b2.len)
    for k, v in t2:
      doAssert $k == v
      doAssert b2.getOrDefault(k) == $k

main()