non underscore group

fjetter · fjetter · commit 03210bc3b2ec · 2021-05-06T22:12:43.000+02:00
diff --git a/distributed/tests/test_scheduler.py b/distributed/tests/test_scheduler.py
@@ -1881,13 +1881,13 @@ async def test_task_groups(c, s, a, b):
     assert tg.nbytes_total == tp.nbytes_total
     # It should map down to individual tasks
     assert tg.nbytes_total == sum(
-        [ts.get_nbytes() for ts in s.tasks.values() if ts._group is tg]
+        [ts.get_nbytes() for ts in s.tasks.values() if ts.group is tg]
     )
     in_memory_ts = sum(
         [
             ts.get_nbytes()
             for ts in s.tasks.values()
-            if ts._group is tg and ts.state == "memory"
+            if ts.group is tg and ts.state == "memory"
         ]
     )
     assert tg.nbytes_in_memory == in_memory_ts

Original file line number	Diff line number	Diff line change
`@@ -1881,13 +1881,13 @@ async def test_task_groups(c, s, a, b):`
`1881`	`1881`	`assert tg.nbytes_total == tp.nbytes_total`
`1882`	`1882`	`# It should map down to individual tasks`
`1883`	`1883`	`assert tg.nbytes_total == sum(`
`1884`		`- [ts.get_nbytes() for ts in s.tasks.values() if ts._group is tg]`
	`1884`	`+ [ts.get_nbytes() for ts in s.tasks.values() if ts.group is tg]`
`1885`	`1885`	`)`
`1886`	`1886`	`in_memory_ts = sum(`
`1887`	`1887`	`[`
`1888`	`1888`	`ts.get_nbytes()`
`1889`	`1889`	`for ts in s.tasks.values()`
`1890`		`- if ts._group is tg and ts.state == "memory"`
	`1890`	`+ if ts.group is tg and ts.state == "memory"`
`1891`	`1891`	`]`
`1892`	`1892`	`)`
`1893`	`1893`	`assert tg.nbytes_in_memory == in_memory_ts`