gradio/test/test_blocks.py

60 lines
1.9 KiB
Python
Raw Normal View History

import random
2022-03-30 00:12:01 +08:00
import unittest
from test.test_data.blocks_configs import XRAY_CONFIG
2022-03-30 20:34:08 +08:00
2022-03-30 01:38:49 +08:00
import gradio as gr
2022-03-30 00:12:01 +08:00
class TestBlocks(unittest.TestCase):
def test_xray(self):
2022-03-29 23:46:42 +08:00
xray_model = lambda diseases, img: {
disease: random.random() for disease in diseases
}
ct_model = lambda diseases, img: {disease: 0.1 for disease in diseases}
xray_blocks = gr.Blocks()
with xray_blocks:
gr.components.Markdown(
"""
# Detect Disease From Scan
With this model you can lorem ipsum
- ipsum 1
- ipsum 2
"""
)
disease = gr.components.CheckboxGroup(
choices=["Covid", "Malaria", "Lung Cancer"], label="Disease to Scan For"
)
with gr.Tabs():
with gr.TabItem("X-ray"):
with gr.Row():
xray_scan = gr.components.Image()
xray_results = gr.components.JSON()
xray_run = gr.Button(
2022-03-29 23:46:42 +08:00
"Run",
css={"background-color": "red", "--hover-color": "orange"},
)
xray_run.click(
xray_model, inputs=[disease, xray_scan], outputs=xray_results
)
with gr.TabItem("CT Scan"):
with gr.Row():
ct_scan = gr.components.Image()
ct_results = gr.components.JSON()
ct_run = gr.Button("Run")
2022-03-29 23:46:42 +08:00
ct_run.click(
ct_model, inputs=[disease, ct_scan], outputs=ct_results
)
2022-03-29 23:46:42 +08:00
_ = gr.components.Textbox()
2022-03-30 01:38:49 +08:00
2022-04-05 08:48:15 +08:00
# print(xray_blocks.get_config_file())
2022-03-30 00:12:01 +08:00
self.assertEqual(XRAY_CONFIG, xray_blocks.get_config_file())
if __name__ == "__main__":
unittest.main()