Skip to content

Commit

Permalink
Replace tabs with spaces in tasks test
Browse files Browse the repository at this point in the history 
How did those tabs get in there?
  • Loading branch information
@WalterKolczynski-NOAA
WalterKolczynski-NOAA committed Jan 8, 2025
1 parent 9372fd2 commit c24412d
Showing 1 changed file with 69 additions and 68 deletions.
137 changes: 69 additions & 68 deletions workflow/tests/test_tasks.py
View file
Original file line number Diff line number Diff line change
@@ -1,83 +1,84 @@
from rocoto.tasks import Tasks


class TestTasks:

'''
Tasks class tests
'''
Tasks class tests
Note: this is currently only testing a small fraction of the class.
'''
Note: this is currently only testing a small fraction of the class.
'''

def test_job_groups(self):
test_array = list(range(0, 24))
def test_job_groups(self):
test_array = list(range(0, 24))

# Test simple splitting with no breakpoints
test_groups = [{'fhrs': [0, 1, 2, 3, 4, 5], 'seg': 0},
{'fhrs': [6, 7, 8, 9, 10, 11], 'seg': 0},
{'fhrs': [12, 13, 14, 15, 16, 17], 'seg': 0},
{'fhrs': [18, 19, 20, 21, 22, 23], 'seg': 0}]
assert Tasks.get_job_groups(fhrs=test_array, ngroups=4) == test_groups
# Test simple splitting with no breakpoints
test_groups = [{'fhrs': [0, 1, 2, 3, 4, 5], 'seg': 0},
{'fhrs': [6, 7, 8, 9, 10, 11], 'seg': 0},
{'fhrs': [12, 13, 14, 15, 16, 17], 'seg': 0},
{'fhrs': [18, 19, 20, 21, 22, 23], 'seg': 0}]
assert Tasks.get_job_groups(fhrs=test_array, ngroups=4) == test_groups

# Test with a break point that aligns with normal split point
test_groups = [{'fhrs': [0, 1, 2, 3, 4, 5], 'seg': 0},
{'fhrs': [6, 7, 8, 9, 10, 11], 'seg': 0},
{'fhrs': [12, 13, 14, 15, 16, 17], 'seg': 1},
{'fhrs': [18, 19, 20, 21, 22, 23], 'seg': 1}]
assert Tasks.get_job_groups(fhrs=test_array, ngroups=4, breakpoints=[11]) == test_groups
# Test with a break point that aligns with normal split point
test_groups = [{'fhrs': [0, 1, 2, 3, 4, 5], 'seg': 0},
{'fhrs': [6, 7, 8, 9, 10, 11], 'seg': 0},
{'fhrs': [12, 13, 14, 15, 16, 17], 'seg': 1},
{'fhrs': [18, 19, 20, 21, 22, 23], 'seg': 1}]
assert Tasks.get_job_groups(fhrs=test_array, ngroups=4, breakpoints=[11]) == test_groups

# Test with a break point not at a normal split point
test_groups = [{'fhrs': [0, 1, 2, 3, 4, 5, 6, 7], 'seg': 0},
{'fhrs': [8, 9, 10, 11, 12, 13, 14], 'seg': 0},
{'fhrs': [15, 16, 17, 18, 19], 'seg': 1},
{'fhrs': [20, 21, 22, 23], 'seg': 1}]
assert Tasks.get_job_groups(fhrs=test_array, ngroups=4, breakpoints=[14]) == test_groups
# Test with a break point not at a normal split point
test_groups = [{'fhrs': [0, 1, 2, 3, 4, 5, 6, 7], 'seg': 0},
{'fhrs': [8, 9, 10, 11, 12, 13, 14], 'seg': 0},
{'fhrs': [15, 16, 17, 18, 19], 'seg': 1},
{'fhrs': [20, 21, 22, 23], 'seg': 1}]
assert Tasks.get_job_groups(fhrs=test_array, ngroups=4, breakpoints=[14]) == test_groups

# Test highly skewed break point
test_groups = [{'fhrs': [0, 1, 2, 3, 4, 5, 6, 7], 'seg': 0},
{'fhrs': [8, 9, 10, 11, 12, 13, 14, 15], 'seg': 0},
{'fhrs': [16, 17, 18, 19, 20, 21, 22], 'seg': 0},
{'fhrs': [23], 'seg': 1}]
assert Tasks.get_job_groups(fhrs=test_array, ngroups=4, breakpoints=[22]) == test_groups
# Test highly skewed break point
test_groups = [{'fhrs': [0, 1, 2, 3, 4, 5, 6, 7], 'seg': 0},
{'fhrs': [8, 9, 10, 11, 12, 13, 14, 15], 'seg': 0},
{'fhrs': [16, 17, 18, 19, 20, 21, 22], 'seg': 0},
{'fhrs': [23], 'seg': 1}]
assert Tasks.get_job_groups(fhrs=test_array, ngroups=4, breakpoints=[22]) == test_groups

# Test with two break points that align
test_groups = [{'fhrs': [0, 1, 2, 3, 4, 5], 'seg': 0},
{'fhrs': [6, 7, 8, 9, 10, 11], 'seg': 0},
{'fhrs': [12, 13, 14, 15, 16, 17], 'seg': 1},
{'fhrs': [18, 19, 20, 21, 22, 23], 'seg': 2}]
assert Tasks.get_job_groups(fhrs=test_array, ngroups=4, breakpoints=[11, 17]) == test_groups
# Test with two break points that align
test_groups = [{'fhrs': [0, 1, 2, 3, 4, 5], 'seg': 0},
{'fhrs': [6, 7, 8, 9, 10, 11], 'seg': 0},
{'fhrs': [12, 13, 14, 15, 16, 17], 'seg': 1},
{'fhrs': [18, 19, 20, 21, 22, 23], 'seg': 2}]
assert Tasks.get_job_groups(fhrs=test_array, ngroups=4, breakpoints=[11, 17]) == test_groups

# Test with two skewed break points
test_groups = [{'fhrs': [0, 1], 'seg': 0},
{'fhrs': [2, 3, 4, 5, 6, 7], 'seg': 1},
{'fhrs': [8, 9, 10, 11, 12], 'seg': 1},
{'fhrs': [13, 14, 15, 16, 17], 'seg': 1},
{'fhrs': [18, 19, 20, 21, 22], 'seg': 1},
{'fhrs': [23], 'seg': 2}]
assert Tasks.get_job_groups(fhrs=test_array, ngroups=6, breakpoints=[1, 22]) == test_groups
# Test with two skewed break points
test_groups = [{'fhrs': [0, 1], 'seg': 0},
{'fhrs': [2, 3, 4, 5, 6, 7], 'seg': 1},
{'fhrs': [8, 9, 10, 11, 12], 'seg': 1},
{'fhrs': [13, 14, 15, 16, 17], 'seg': 1},
{'fhrs': [18, 19, 20, 21, 22], 'seg': 1},
{'fhrs': [23], 'seg': 2}]
assert Tasks.get_job_groups(fhrs=test_array, ngroups=6, breakpoints=[1, 22]) == test_groups

# Test slightly irregular break points
test_groups = [{'fhrs': [0, 1, 2, 3], 'seg': 0},
{'fhrs': [4, 5, 6], 'seg': 0},
{'fhrs': [7, 8, 9, 10], 'seg': 1},
{'fhrs': [11, 12, 13, 14], 'seg': 1},
{'fhrs': [15, 16, 17, 18], 'seg': 1},
{'fhrs': [19, 20, 21, 22, 23], 'seg': 2}]
assert Tasks.get_job_groups(fhrs=test_array, ngroups=6, breakpoints=[6, 18]) == test_groups
# Test slightly irregular break points
test_groups = [{'fhrs': [0, 1, 2, 3], 'seg': 0},
{'fhrs': [4, 5, 6], 'seg': 0},
{'fhrs': [7, 8, 9, 10], 'seg': 1},
{'fhrs': [11, 12, 13, 14], 'seg': 1},
{'fhrs': [15, 16, 17, 18], 'seg': 1},
{'fhrs': [19, 20, 21, 22, 23], 'seg': 2}]
assert Tasks.get_job_groups(fhrs=test_array, ngroups=6, breakpoints=[6, 18]) == test_groups

# Test more groups than fhrs available
test_array = list(range(0, 6))
test_groups = [{'fhrs': [0], 'seg': 0},
{'fhrs': [1], 'seg': 0},
{'fhrs': [2], 'seg': 0},
{'fhrs': [3], 'seg': 0},
{'fhrs': [4], 'seg': 0},
{'fhrs': [5], 'seg': 0}]
assert Tasks.get_job_groups(fhrs=test_array, ngroups=15) == test_groups
# Test more groups than fhrs available
test_array = list(range(0, 6))
test_groups = [{'fhrs': [0], 'seg': 0},
{'fhrs': [1], 'seg': 0},
{'fhrs': [2], 'seg': 0},
{'fhrs': [3], 'seg': 0},
{'fhrs': [4], 'seg': 0},
{'fhrs': [5], 'seg': 0}]
assert Tasks.get_job_groups(fhrs=test_array, ngroups=15) == test_groups

def test_multiply_HMS(self):
assert Tasks.multiply_HMS('00:10:00', 2) == '00:20:00'
assert Tasks.multiply_HMS('00:30:00', 10) == '05:00:00'
assert Tasks.multiply_HMS('01:15:00', 4) == '05:00:00'
assert Tasks.multiply_HMS('00:05:00', 1.5) == '00:07:30'
assert Tasks.multiply_HMS('00:40:00', 2.5) == '01:40:00'
assert Tasks.multiply_HMS('00:10:00', 1) == '00:10:00'
def test_multiply_HMS(self):
assert Tasks.multiply_HMS('00:10:00', 2) == '00:20:00'
assert Tasks.multiply_HMS('00:30:00', 10) == '05:00:00'
assert Tasks.multiply_HMS('01:15:00', 4) == '05:00:00'
assert Tasks.multiply_HMS('00:05:00', 1.5) == '00:07:30'
assert Tasks.multiply_HMS('00:40:00', 2.5) == '01:40:00'
assert Tasks.multiply_HMS('00:10:00', 1) == '00:10:00'

0 comments on commit c24412d

Please sign in to comment.