Spaces:

omerXfaruq
/

Model-Comparator-Space-Builder

Build error

Remove spaces hardcoding, and support models as well.

b8ac61b over 2 years ago

No virus

1.52 kB

	import pytest
	from app import SpaceBuilder


	class TestUnit:
	class TestSpaceBuilder:
	def test_split_space_names(self):
	input = (
	f"spaces/nielsr/LayoutLMv2-FUNSD"
	f"\nspaces/valhalla/glide-text2im"
	f"\n"
	f"\n "
	f"\n "
	f"\nspaces/valhalla/glide-text2im"
	f"\n "
	f"\nspaces/valhalla/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"spaces/nielsr/LayoutLMv2-FUNSD"
	f"\nspaces/valhalla/glide-text2im"
	)
	assert SpaceBuilder.load_and_check_spaces(names=input) is False

	def test_load_and_check_spaces_2(self):
	input = (
	f"spaces/valhalla/glide-text2im"
	f"\nspaces/valhalla/glide-text2im"
	)
	assert SpaceBuilder.load_and_check_spaces(names=input) is True