File: test_task_id.py

package info (click to toggle)
cylc-flow 8.6.1-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 14,368 kB
  • sloc: python: 87,751; sh: 17,109; sql: 233; xml: 171; javascript: 78; lisp: 55; makefile: 11
file content (67 lines) | stat: -rw-r--r-- 2,458 bytes parent folder | download | duplicates (3)
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
# THIS FILE IS PART OF THE CYLC WORKFLOW ENGINE.
# Copyright (C) NIWA & British Crown (Met Office) & Contributors.
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program.  If not, see <http://www.gnu.org/licenses/>.

import unittest

from cylc.flow.task_id import TaskID


class TestTaskId(unittest.TestCase):

    def test_get(self):
        self.assertEqual("a.1", TaskID.get("a", 1))
        self.assertEqual("a._1", TaskID.get("a", "_1"))
        self.assertEqual(
            "WTASK.20101010T101010", TaskID.get("WTASK", "20101010T101010"))

    def test_split(self):
        self.assertEqual(["a", '1'], TaskID.split("a.1"))
        self.assertEqual(["a", '_1'], TaskID.split("a._1"))
        self.assertEqual(
            ["WTAS", '20101010T101010'], TaskID.split("WTAS.20101010T101010"))

    def test_is_valid_name(self):
        for name in [
            "abc", "123", "____", "_", "a_b", "a_1", "1_b", "ABC"
        ]:
            self.assertTrue(TaskID.is_valid_name(name))
        for name in [
            "a.1", None, "%abc", "", " "
        ]:
            self.assertFalse(TaskID.is_valid_name(name))

    def test_is_valid_id(self):
        for id1 in [
            "a.1", "_.098098439535$#%#@!#~"
        ]:
            self.assertTrue(TaskID.is_valid_id(id1))
        for id2 in [
            "abc", "123", "____", "_", "a_b", "a_1", "1_b", "ABC", "a.A A"
        ]:
            self.assertFalse(TaskID.is_valid_id(id2))

    def test_is_valid_id_2(self):
        # TBD: a.A A is invalid for valid_id, but valid for valid_id_2?
        # TBD: a/a.a is OK?
        for id1 in [
            "a.1", "_.098098439535$#%#@!#~", "a/1", "_/098098439535$#%#@!#~",
            "a.A A", "a/a.a"
        ]:
            self.assertTrue(TaskID.is_valid_id_2(id1))
        for id2 in [
            "abc", "123", "____", "_", "a_b", "a_1", "1_b", "ABC"
        ]:
            self.assertFalse(TaskID.is_valid_id_2(id2))