Skip to content
Snippets Groups Projects

Add test for format command

Merged Yoann Schneider requested to merge unit-tests-for-format into main
1 file
+ 36
0
Compare changes
  • Side-by-side
  • Inline
+ 36
0
# -*- coding: utf-8 -*-
from pathlib import Path
import pytest
from dan.datasets.format.atr import ATRDatasetFormatter, remove_spaces
@pytest.mark.parametrize(
"text,trimmed",
(
("no_spaces", "no_spaces"),
(" beginning", "beginning"),
("ending ", "ending"),
(" both ", "both"),
(" consecutive", "consecutive"),
("\ttab", "tab"),
),
)
def test_remove_spaces(text, trimmed):
assert remove_spaces(text) == trimmed
@pytest.mark.parametrize(
"image_format, expected_format",
(
(".png", "png"),
("png", "png"),
),
)
def test_image_format_parsing(image_format, expected_format):
formatter = ATRDatasetFormatter(
dataset=Path("my_dataset"), image_format=image_format, remove_spaces=True
)
assert formatter.image_format == expected_format
Loading