Spaces:
Build error
Build error
| import pytest | |
| from app import SpaceBuilder | |
| class TestUnit: | |
| class TestSpaceBuilder: | |
| def test_split_space_names(self): | |
| input = ( | |
| f"nielsr/LayoutLMv2-FUNSD" | |
| f"\nvalhalla/glide-text2im" | |
| f"\n" | |
| f"\n " | |
| f"\n " | |
| f"\nvalhalla/glide-text2im" | |
| f"\n " | |
| f"\nvalhalla/glide-text2im" | |
| f"\n" | |
| ) | |
| expected_output = ( | |
| ["spaces/nielsr/LayoutLMv2-FUNSD", "spaces/valhalla/glide-text2im", "spaces/valhalla/glide-text2im", "spaces/valhalla/glide-text2im"] | |
| ) | |
| assert expected_output == SpaceBuilder.split_space_names(input) | |
| def test_check_space_name_availability(self): | |
| random_token = "123123" | |
| random_space_name = "123123" | |
| assert SpaceBuilder.check_space_name_availability(random_token, random_space_name) is False | |
| def test_load_and_check_spaces_1(self): | |
| input = ( | |
| f"nielsr/LayoutLMv2-FUNSD" | |
| f"\nvalhalla/glide-text2im" | |
| ) | |
| assert SpaceBuilder.load_and_check_spaces(names=input) is False | |
| def test_load_and_check_spaces_2(self): | |
| input = ( | |
| f"valhalla/glide-text2im" | |
| f"\nvalhalla/glide-text2im" | |
| ) | |
| assert SpaceBuilder.load_and_check_spaces(names=input) is True | |