aqmt-example/example.py at master · henrist/aqmt-example · GitHub

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
#!/usr/bin/env python3
#
# This is an example of how to use the AQM test framework.
#
# Usage:
# ./example.py
#

import sys

from aqmt import Testbed, TestEnv, run_test, steps
from aqmt.plot import collection_components, flow_components
from aqmt.traffic import greedy


def test(result_folder):

    def my_test(testcase):
        testcase.traffic(greedy, node='a', tag='CUBIC')
        testcase.traffic(greedy, node='b', tag='ECN-CUBIC')

    testbed = Testbed()

    testbed.ta_samples = 50
    testbed.ta_delay = 250

    testbed.cc('a', 'cubic', testbed.ECN_ALLOW)
    testbed.cc('b', 'cubic', testbed.ECN_INITIATE)

    run_test(
        folder=result_folder,
        title='Just a simple test to demonstrate usage',
        testenv=TestEnv(testbed),
        steps=(
            steps.plot_compare(level_order=[], components=[
                collection_components.utilization_tags(),
                collection_components.window_rate_ratio(),
                collection_components.queueing_delay(),
                collection_components.queueing_delay(y_logarithmic=True),
                collection_components.drops_marks(),
                collection_components.drops_marks(y_logarithmic=True),
            ]),
            steps.plot_flows(level_order=[], components=[
                flow_components.utilization_queues(),
                flow_components.rate_per_flow(),
                flow_components.rate_per_flow(y_logarithmic=True),
                flow_components.window(),
                flow_components.window(y_logarithmic=True),
                flow_components.queueing_delay(),
                flow_components.queueing_delay(y_logarithmic=True),
                flow_components.drops_marks(),
                flow_components.drops_marks(y_logarithmic=True),
            ]),
            steps.branch_sched([
                # tag, title, name, params
                ('pie', 'PIE', 'pie', 'ecn'),
                ('fq_codel', 'fq\\\\_codel', 'fq_codel', ''),
                ('pfifo', 'pfifo', 'pfifo', ''),
            ]),
            steps.branch_bitrate([
                10,
                50,
            ]),
            steps.branch_rtt([
                2,
                10,
                50,
            ], title='%d'),
            my_test,
        )
    )

if __name__ == '__main__':
    result_folder = 'results/example'
    if len(sys.argv) >= 2:
        result_folder = sys.argv[1]

    test(result_folder)