def test_Append(self):
example_list = utilities.create_example_daglist()
new_object = c4d.BaseList2D(c4d.Onull)
new_object.SetName("NewItem_Null")
example_list.Append(dag.DagAtom(new_object))
result = example_list.Get("NewItem_Null").GetName()
result_expected = "NewItem_Null"
self.assertEqual(result, result_expected)
def test_Insert(self):
example_list = utilities.create_example_daglist()
new_object = c4d.BaseList2D(c4d.Onull)
new_object.SetName("NewItem_Null")
example_list.Insert(0, dag.DagAtom(new_object))
result = [x.GetName() for x in example_list]
result_expected = ["NewItem_Null"
] + [f"BaseObject_{x}_Null" for x in range(1, 11)]
self.assertListEqual(result, result_expected)
def test_Get(self):
example_list = utilities.create_example_daglist()
item = example_list.Get("BaseObject_2_Null")
# assert resulting instance
self.assertIsInstance(item, dag.DagAtom)
# assert resulting name
result = item.GetName()
result_expected = "BaseObject_2_Null"
self.assertEqual(result, result_expected)
def test___iter__(self):
example_list = utilities.create_example_daglist()
result = []
for item in example_list:
result.append(item.GetName())
# assert resulting instance
self.assertIsInstance(item, dag.DagAtom)
result_expected = [
"BaseObject_{}_Null".format(x + 1) for x in range(0, 10)
]
self.assertListEqual(result, result_expected)
def test_CleanUp(self):
example_list = utilities.create_example_daglist()
example_list[0].Remove()
# before clean up
result_before_cleanup = [x.GetName() for x in example_list._items]
result_before_cleanup_expected = [
f"BaseObject_{x}_Null" for x in range(1, 11)
]
self.assertListEqual(result_before_cleanup,
result_before_cleanup_expected)
example_list.CleanUp()
# after clean up
result_after_cleanup = [x.GetName() for x in example_list._items]
result_after_cleanup_expected = [
f"BaseObject_{x}_Null" for x in range(2, 11)
]
self.assertListEqual(result_after_cleanup,
result_after_cleanup_expected)
def test_Get_raise_DagNotFoundError(self):
example_list = utilities.create_example_daglist()
with self.assertRaises(dag.DagNotFoundError):
example_list.Get("YouCanNotFindMe")