mirror of
https://github.com/clearml/clearml-server
synced 2025-06-26 23:15:47 +00:00
Add organization.update_company_name
Fix unit-tests
This commit is contained in:
@@ -2,9 +2,6 @@ from apiserver.tests.automated import TestService
|
||||
|
||||
|
||||
class TestOrganization(TestService):
|
||||
def setUp(self, version="2.12"):
|
||||
super().setUp(version=version)
|
||||
|
||||
def test_get_user_companies(self):
|
||||
company = self.api.organization.get_user_companies().companies[0]
|
||||
self.assertEqual(len(company.owners), company.allocated)
|
||||
|
||||
@@ -2,7 +2,7 @@ import math
|
||||
from apiserver.tests.automated import TestService
|
||||
|
||||
|
||||
class TestEntityOrdering(TestService):
|
||||
class TestPagingAndScrolling(TestService):
|
||||
name_prefix = f"Test paging "
|
||||
|
||||
def setUp(self, **kwargs):
|
||||
@@ -13,7 +13,16 @@ class TestEntityOrdering(TestService):
|
||||
tasks = [
|
||||
self._temp_task(
|
||||
name=f"{self.name_prefix}{i}",
|
||||
hyperparams={"test": {"param": {"section": "test", "name": "param", "type": "str", "value": str(i)}}},
|
||||
hyperparams={
|
||||
"test": {
|
||||
"param": {
|
||||
"section": "test",
|
||||
"name": "param",
|
||||
"type": "str",
|
||||
"value": str(i),
|
||||
}
|
||||
}
|
||||
},
|
||||
)
|
||||
for i in range(18)
|
||||
]
|
||||
@@ -24,10 +33,7 @@ class TestEntityOrdering(TestService):
|
||||
for page in range(0, math.ceil(len(self.task_ids) / page_size)):
|
||||
start = page * page_size
|
||||
expected_size = min(page_size, len(self.task_ids) - start)
|
||||
tasks = self._get_tasks(
|
||||
page=page,
|
||||
page_size=page_size,
|
||||
).tasks
|
||||
tasks = self._get_tasks(page=page, page_size=page_size,).tasks
|
||||
self.assertEqual(len(tasks), expected_size)
|
||||
for i, t in enumerate(tasks):
|
||||
self.assertEqual(t.name, f"{self.name_prefix}{start + i}")
|
||||
@@ -38,10 +44,7 @@ class TestEntityOrdering(TestService):
|
||||
for page in range(0, math.ceil(len(self.task_ids) / page_size)):
|
||||
start = page * page_size
|
||||
expected_size = min(page_size, len(self.task_ids) - start)
|
||||
res = self._get_tasks(
|
||||
size=page_size,
|
||||
scroll_id=scroll_id,
|
||||
)
|
||||
res = self._get_tasks(size=page_size, scroll_id=scroll_id,)
|
||||
self.assertTrue(res.scroll_id)
|
||||
scroll_id = res.scroll_id
|
||||
tasks = res.tasks
|
||||
@@ -50,24 +53,19 @@ class TestEntityOrdering(TestService):
|
||||
self.assertEqual(t.name, f"{self.name_prefix}{start + i}")
|
||||
|
||||
# no more data in this scroll
|
||||
tasks = self._get_tasks(
|
||||
size=page_size,
|
||||
scroll_id=scroll_id,
|
||||
).tasks
|
||||
tasks = self._get_tasks(size=page_size, scroll_id=scroll_id,).tasks
|
||||
self.assertFalse(tasks)
|
||||
|
||||
# refresh brings all
|
||||
tasks = self._get_tasks(
|
||||
size=page_size,
|
||||
scroll_id=scroll_id,
|
||||
refresh_scroll=True,
|
||||
size=page_size, scroll_id=scroll_id, refresh_scroll=True,
|
||||
).tasks
|
||||
self.assertEqual([t.id for t in tasks], self.task_ids)
|
||||
|
||||
def _get_tasks(self, **page_params):
|
||||
return self.api.tasks.get_all_ex(
|
||||
name="^Test paging ",
|
||||
order_by=["hyperparams.param"],
|
||||
order_by=["hyperparams.test.param.value"],
|
||||
**page_params,
|
||||
)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user