Skip to content
Snippets Groups Projects
Commit 80771a36 authored by Yoann Schneider's avatar Yoann Schneider :tennis: Committed by Solene Tarride
Browse files

Add test for format command

parent c8e21cb9
No related branches found
No related tags found
1 merge request!28Add test for format command
# -*- 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
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment