Skip to content

Commit 8e78125

Browse files
committed
test: Fix GPU tests
1 parent 69c5438 commit 8e78125

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

test.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -164,12 +164,12 @@ def test_cpueff(db):
164164

165165
def test_gpueff(db):
166166
data = """
167-
JobID,AllocTRES,TRESUsageInAve
167+
JobID,AllocTRES,TRESUsageInTot
168168
1,gres/gpu=1,gres/gpuutil=23
169169
"""
170170
slurm2sql.slurm2sql(db, [], csv_input=csvdata(data))
171171
print(db.execute('select * from eff;').fetchall())
172-
assert fetch(db, 1, 'GPUEff', table='eff') == 0.23
172+
assert fetch(db, 1, 'GpuEff', table='eff') == 0.23
173173

174174

175175
#
@@ -242,7 +242,7 @@ def test_seff_mem(db, capsys):
242242

243243
def test_seff_gpu(db, capsys):
244244
data = """
245-
JobID,End,Elapsed,TotalCPU,NCPUS,AllocTRES,TRESUsageInAve
245+
JobID,End,Elapsed,TotalCPU,NCPUS,AllocTRES,TRESUsageInTot
246246
111,1970-01-01T00:00:00,,1,1,,
247247
111.2,1970-01-01T00:00:00,100,1,1,gres/gpu=1,gres/gpuutil=23
248248
"""

0 commit comments

Comments
 (0)