File: py_u_TestAddNodeFunc.py

package info (click to toggle)
ecflow 5.15.2-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 51,868 kB
  • sloc: cpp: 269,341; python: 22,756; sh: 3,609; perl: 770; xml: 333; f90: 204; ansic: 141; makefile: 70
file content (67 lines) | stat: -rw-r--r-- 2,339 bytes parent folder | download
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
#
# Copyright 2009- ECMWF.
#
# This software is licensed under the terms of the Apache Licence version 2.0
# which can be obtained at http://www.apache.org/licenses/LICENSE-2.0.
# In applying this licence, ECMWF does not waive the privileges and immunities
# granted to it by virtue of its status as an intergovernmental organisation
# nor does it submit to any jurisdiction.
#

from ecflow import Defs, Client, debug_build
import ecflow_test_util as Test
import os


def create_defs_sequentially():
    local_defs = Defs()
    suite = local_defs.add_suite("s1")
    suite.add_variable("Var", "value")
    suite.add_meter("metername3", 0, 100, 50)
    suite.add_event(2)
    suite.add_event(3)
    f1 = suite.add_family("f1")
    f1 = f1.add_family("f1")
    t1 = f1.add_task("t1")
    t1.add_variable("fred", "jones")

    f2 = suite.add_family("f2")
    f1 = f2.add_family("f1")
    t1 = f1.add_task("t1")
    t1.add_variable("fred", "jones")

    f3 = suite.add_family("f3")
    t1 = f3.add_task("t1")
    t1.add_variable("fred", "jones")

    suite.add_task("t1").add_variable("fred", "jones")
    return local_defs


def create_defs_functionally():
    f_defs = Defs()
    suite = f_defs.add_suite("s1")
    suite.add_variable("Var", "value").add_family("f1").add_family("f1").add_task("t1").add_variable("fred", "jones")
    suite.add_family("f2").add_family("f1").add_task("t1").add_variable("fred", "jones")
    suite.add_meter("metername3", 0, 100, 50).add_family("f3").add_task("t1").add_variable("fred", "jones")
    suite.add_event(2).add_event(3).add_task("t1").add_variable("fred", "jones")
    return f_defs


if __name__ == "__main__":
    Test.print_test_start(os.path.basename(__file__))

    #
    # Add Nodes functional way
    #
    defs = Defs()
    defs.add_suite("s1").add_task("t1").add_variable("var", "v")
    defs.add_suite("s2").add_family("f1").add_task("t1").add_variable("var", "v")
    defs.add_suite("s3").add_family("f1").add_family("f2").add_task("t1").add_variable("var", "v")
    assert len(defs) == 3, "Expected 3 suites"

    # These test show that although add_variable() returns a node_ptr, its really a suite
    # hence calling add_family/add_task still works.
    assert create_defs_sequentially() == create_defs_functionally(), "Functional suite not as expected "

    print("All Tests pass")