def test_load_sheet_from_django_model(self):
model=FakeDjangoModel()
importer = DjangoModelImporter()
adapter = DjangoModelImportAdapter(model)
adapter.set_column_names(self.data[0])
importer.append(adapter)
save_data(importer, {adapter.get_name(): self.data[1:]}, file_type=DB_DJANGO)
assert model.objects.objs == self.result
model._meta.update(["X", "Y", "Z"])
reader = DjangoModelReader(model)
data = reader.to_array()
assert list(data) == self.data
def test_load_sheet_from_django_model(self):
model = FakeDjangoModel()
importer = DjangoModelImporter()
adapter = DjangoModelImportAdapter(model)
adapter.column_names = self.data[0]
importer.append(adapter)
save_data(importer, {adapter.get_name(): self.data[1:]},
file_type=DB_DJANGO)
assert model.objects.objs == self.result
model._meta.update(["X", "Y", "Z"])
reader = DjangoModelReader(model)
data = reader.to_array()
assert list(data) == self.data
def test_load_and_format_sheet_from_django_model(self):
model = FakeDjangoModel()
importer = DjangoModelImporter()
adapter = DjangoModelImportAdapter(model)
adapter.column_names = self.data[0]
importer.append(adapter)
save_data(importer, {adapter.get_name(): self.data[1:]},
file_type=DB_DJANGO)
assert model.objects.objs == self.result
model._meta.update(["X", "Y", "Z"])
def row_renderer(row):
return [str(element) for element in row]
# the key point of this test case
reader = DjangoModelReader(model, row_renderer=row_renderer)
data = reader.to_array()
expected = [["X", "Y", "Z"], ['1', '2', '3'], ['4', '5', '6']]
eq_(list(data), expected)
def test_load_and_format_sheet_from_django_model(self):
model = FakeDjangoModel()
importer = DjangoModelImporter()
adapter = DjangoModelImportAdapter(model)
adapter.column_names = self.data[0]
importer.append(adapter)
save_data(importer, {adapter.get_name(): self.data[1:]},
file_type=DB_DJANGO)
assert model.objects.objs == self.result
model._meta.update(["X", "Y", "Z"])
def row_renderer(row):
return [str(element) for element in row]
# the key point of this test case
reader = DjangoModelReader(model,
row_renderer=row_renderer)
data = reader.to_array()
expected = [
["X", "Y", "Z"],
['1', '2', '3'],
['4', '5', '6']
]
eq_(list(data), expected)
def test_load_sheet_from_django_model_with_filter_3(self):
reader = DjangoModelReader(self.model, start_column=1, column_limit=1)
data = reader.to_array()
expected = [['Y'], [2], [5]]
eq_(list(data), expected)
def test_load_sheet_from_django_model_with_filter_1(self):
reader = DjangoModelReader(self.model, start_row=1, row_limit=3)
data = reader.to_array()
expected = [[1, 2, 3], [4, 5, 6]]
eq_(list(data), expected)
def test_load_sheet_from_django_model_with_filter(self):
reader = DjangoModelReader(self.model, start_row=0, row_limit=2)
data = reader.to_array()
expected = [['X', 'Y', 'Z'], [1, 2, 3]]
eq_(list(data), expected)
def test_empty_model(self):
model = FakeDjangoModel()
reader = DjangoModelReader(model)
data = reader.to_array()
eq_(list(data), [[], []])
def test_load_sheet_from_django_model_with_filter_3(self):
reader = DjangoModelReader(self.model, start_column=1, column_limit=1)
data = reader.to_array()
expected = [['Y'], [2], [5]]
eq_(list(data), expected)
def test_load_sheet_from_django_model_with_filter_1(self):
reader = DjangoModelReader(self.model, start_row=1, row_limit=3)
data = reader.to_array()
expected = [[1, 2, 3], [4, 5, 6]]
eq_(list(data), expected)
def test_load_sheet_from_django_model_with_filter(self):
reader = DjangoModelReader(self.model, start_row=0, row_limit=2)
data = reader.to_array()
expected = [['X', 'Y', 'Z'], [1, 2, 3]]
eq_(list(data), expected)
def test_empty_model(self):
model = FakeDjangoModel()
reader = DjangoModelReader(model)
data = reader.to_array()
eq_(list(data), [[]])
def test_empty_model(self):
model = FakeDjangoModel()
reader = DjangoModelReader(model)
data = reader.to_array()
assert data == []