summary refs log tree commit diff stats
path: root/tests/stdlib/tsortcall.nim
blob: 242e3fe4c5adb7349111e75c6872fc6f638a624c (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
discard """
outputsub: ""
"""

import algorithm
import unittest


suite "test sort, sorted, and isSorted procs":
  proc foosort(ships: var seq[int]) = sort(ships, system.cmp[int])

  type
    User = object
      name: string
      age: int

  func newUser(name: string, age: int): User =
    result.name = name
    result.age = age

  proc compareUsers(x, y: User): int =
    if x.age == y.age: return 0
    if x.age < y.age: return -1
    return 1

  setup:
    var
      unSortedIntSeq = @[1, 4, 3, 5, -1]
      unSortedUserSeq = @[newUser("Andreas", 34), newUser("Alice", 12), newUser("Bob", 23)]

    let
      sortedIntSeq = @[-1, 1, 3, 4, 5]
      sortedUserSeq = @[newUser("Alice", 12), newUser("Bob", 23), newUser("Andreas", 34)]

  test "test the shortcut versions of sort, sorted, and isSorted":
    check(not unSortedIntSeq.isSorted)
    check sorted(unSortedIntSeq) == sortedIntSeq
    check sorted(unSortedIntSeq).isSorted

    unSortedIntSeq.sort()
    check unSortedIntSeq == sortedIntSeq
    check unSortedIntSeq.isSorted

  test "test the shortcut versions with descending sort order":
    check(not unSortedIntSeq.isSorted(SortOrder.Descending))
    check sorted(unSortedIntSeq, SortOrder.Descending) == reversed sortedIntSeq
    check sorted(unSortedIntSeq).isSorted(SortOrder.Ascending)

    unSortedIntSeq.sort(SortOrder.Descending)
    check unSortedIntSeq == reversed sortedIntSeq
    check unSortedIntSeq.isSorted(SortOrder.Descending)

  test "test the versions that accept a custom compareUsers function":
    check(not unSortedUserSeq.isSorted(compareUsers))
    check sorted(unSortedUserSeq, compareUsers) == sortedUserSeq
    check sorted(unSortedUserSeq, compareUsers).isSorted(compareUsers)

    unSortedUserSeq.sort(compareUsers)
    check unSortedUserSeq == sortedUserSeq
    check unSortedUserSeq.isSorted(compareUsers)

  test "test the long versions with descending sort order":
    check(not unSortedUserSeq.isSorted(compareUsers, SortOrder.Descending))
    check sorted(unSortedUserSeq, compareUsers, SortOrder.Descending) == reversed sortedUserSeq
    check sorted(unSortedUserSeq, compareUsers,
                  SortOrder.Descending).isSorted(compareUsers, SortOrder.Descending)
    unSortedUserSeq.sort(compareUsers, SortOrder.Descending)
    check unSortedUserSeq == reversed sortedUserSeq
    check unSortedUserSeq.isSorted(compareUsers, SortOrder.Descending)