Python Dataset._get_headers示例

示例#1

def test_friendly_renderer():
    user = UserFactory(is_superuser=True)
    BookFactory()
    BookFactory(best_seller=True)

    client = APIClient()
    client.force_authenticate(user=user)
    response = client.get(reverse('sample:book-csv-view') + '?format=csv')
    dataset = Dataset().load(response.content.decode('utf-8'), 'csv')
    assert len(dataset._get_headers()) == 4
    assert dataset[0][2] == ''
    assert dataset[1][2] == 'Yes'

示例#2

文件： test_views.py 项目： unicef/unicef-rest-export

def test_export_model_view():
    user = UserFactory(is_superuser=True)
    BookFactory(name='demo')
    BookFactory(name='test')

    client = APIClient()
    client.force_authenticate(user=user)

    url = reverse('sample:author-list') + '?format=csv'
    response = client.get(url, format='json')

    dataset = Dataset().load(response.content.decode('utf-8'), 'csv')
    assert len(dataset._get_headers()) == 4
    assert dataset.headers == ['ID', 'Books', 'First name', 'Last name']