mirror of
https://github.com/gradio-app/gradio.git
synced 2024-12-21 02:19:59 +08:00
fix failing tests
This commit is contained in:
parent
bcb3c37a3b
commit
7ac6abe909
@ -75,7 +75,7 @@ h4 {
|
||||
cursor: pointer;
|
||||
}
|
||||
.examples > button {
|
||||
padding: 4px;
|
||||
padding: 8px 16px;
|
||||
border-radius: 2px;
|
||||
margin-right: 4px;
|
||||
background-color: whitesmoke;
|
||||
|
@ -44,4 +44,4 @@ io = gr.Interface(
|
||||
]
|
||||
)
|
||||
|
||||
io.launch(share=True)
|
||||
io.launch()
|
@ -75,7 +75,7 @@ h4 {
|
||||
cursor: pointer;
|
||||
}
|
||||
.examples > button {
|
||||
padding: 4px;
|
||||
padding: 8px 16px;
|
||||
border-radius: 2px;
|
||||
margin-right: 4px;
|
||||
background-color: whitesmoke;
|
||||
|
@ -10,7 +10,7 @@ class TestDefault(unittest.TestCase):
|
||||
def test_default_text(self):
|
||||
max_word_len = lambda text: max([len(word) for word in text.split(" ")])
|
||||
text_interface = Interface(max_word_len, "textbox", "label", interpretation="default")
|
||||
interpretation = text_interface.interpret(["quickest brown fox"])[0]
|
||||
interpretation = text_interface.interpret(["quickest brown fox"])[0][0]
|
||||
self.assertGreater(interpretation[0][1], 0) # Checks to see if the first word has >0 score.
|
||||
self.assertEqual(interpretation[-1][1], 0) # Checks to see if the last word has 0 score.
|
||||
|
||||
@ -20,7 +20,7 @@ class TestDefault(unittest.TestCase):
|
||||
array = np.zeros((100,100))
|
||||
array[0, 0] = 1
|
||||
img = encode_array_to_base64(array)
|
||||
interpretation = img_interface.interpret([img])[0]
|
||||
interpretation = img_interface.interpret([img])[0][0]
|
||||
self.assertGreater(interpretation[0][0], 0) # Checks to see if the top-left has >0 score.
|
||||
|
||||
|
||||
@ -29,14 +29,14 @@ class TestCustom(unittest.TestCase):
|
||||
max_word_len = lambda text: max([len(word) for word in text.split(" ")])
|
||||
custom = lambda text: [(char, 1) for char in text]
|
||||
text_interface = Interface(max_word_len, "textbox", "label", interpretation=custom)
|
||||
result = text_interface.interpret(["quickest brown fox"])[0]
|
||||
result = text_interface.interpret(["quickest brown fox"])[0][0]
|
||||
self.assertEqual(result[0][1], 1) # Checks to see if the first letter has score of 1.
|
||||
|
||||
def test_custom_img(self):
|
||||
max_pixel_value = lambda img: img.max()
|
||||
custom = lambda img: img.tolist()
|
||||
img_interface = Interface(max_pixel_value, "image", "label", interpretation=custom)
|
||||
result = img_interface.interpret([gradio.test_data.BASE64_IMAGE])[0]
|
||||
result = img_interface.interpret([gradio.test_data.BASE64_IMAGE])[0][0]
|
||||
expected_result = np.asarray(decode_base64_to_image(gradio.test_data.BASE64_IMAGE).convert('RGB')).tolist()
|
||||
self.assertEqual(result, expected_result)
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user