如何为返回列表或数组的方法编写单元测试

from django.db import models

class Task(models.Model):
    title = models.CharField(max_length=100, blank=True)
    desc = models.TextField(blank=True)
    tags = models.TextField(null=True, blank=True)

    def tag_list(self):
        """
        将tags字段的逗号分隔字符串转换为标签列表。
        """
        if not self.tags:
            return []
        # 使用strip()去除每个标签可能存在的空白字符
        return [tag.strip() for tag in self.tags.split(',')]

    def __str__(self):
        return self.title

from django.test import TestCase
# 假设Task模型位于'your_app_name.models'
from your_app_name.models import Task 

class TaskModelTest(TestCase):
    def test_tag_list_method(self):
        """
        测试Task模型的tag_list方法是否能正确返回标签列表。
        """
        # 1. 准备测试数据
        # 创建一个Task实例，其tags字段包含逗号分隔的字符串
        task_data = {
            "title": "测试任务",
            "desc": "这是一个测试描述。",
            "tags": "python, django, unittest"
        }
        task = Task.objects.create(**task_data)

        # 2. 调用待测试方法
        actual_tags = task.tag_list()

        # 3. 定义预期结果
        expected_tags = ['python', 'django', 'unittest']

        # 4. 断言：比较实际结果与预期结果
        self.assertEqual(actual_tags, expected_tags)

# your_app_name/tests/test_models.py

from django.test import TestCase
from your_app_name.models import Task # 替换为你的实际应用名

class TaskModelTest(TestCase):
    def test_tag_list_method_with_multiple_tags(self):
        """
        测试tag_list方法处理多个标签的场景。
        """
        task = Task.objects.create(
            title="学习Python",
            desc="学习Django和单元测试",
            tags="python,django,unittest"
        )
        self.assertEqual(task.tag_list(), ['python', 'django', 'unittest'])

    def test_tag_list_method_with_single_tag(self):
        """
        测试tag_list方法处理单个标签的场景。
        """
        task = Task.objects.create(
            title="单个标签任务",
            desc="只有一个标签",
            tags="programming"
        )
        self.assertEqual(task.tag_list(), ['programming'])

    def test_tag_list_method_with_empty_tags(self):
        """
        测试tag_list方法处理空标签字符串的场景。
        """
        task = Task.objects.create(
            title="无标签任务",
            desc="这个任务没有标签",
            tags=""
        )
        self.assertEqual(task.tag_list(), [])

        task_null = Task.objects.create(
            title="空值标签任务",
            desc="这个任务的标签是null",
            tags=None
        )
        self.assertEqual(task_null.tag_list(), [])

    def test_tag_list_method_with_spaces_in_tags(self):
        """
        测试tag_list方法处理标签中包含空格的场景，确保strip()生效。
        """
        task = Task.objects.create(
            title="带空格标签任务",
            desc="标签中包含额外空格",
            tags="  python ,  django , unittest "
        )
        self.assertEqual(task.tag_list(), ['python', 'django', 'unittest'])