3dai

Running on Zero

3dai / tests /test_preview_headers.py

Bobby

Preserve mobile preview resolution across Gradio GPU requests

a8b1bfd about 2 months ago

782 Bytes

import pickle
import unittest

from tests.test_glb_cache import app, gr


class PreviewRequestHeaders(unittest.TestCase):
    def test_mobile_resolution_survives_request_serialization(self):
        for header in ("user-agent", "User-Agent"):
            request = gr.Request(headers={header: "Mozilla/5.0 (iPhone) Mobile"})
            restored = pickle.loads(pickle.dumps(request))
            self.assertEqual(app.get_preview_settings(restored), (True, 256, 15, 3))

    def test_desktop_and_missing_requests_keep_desktop_resolution(self):
        request = gr.Request(headers={"user-agent": "Mozilla/5.0 (Macintosh)"})
        self.assertEqual(app.get_preview_settings(request), (False, 384, 15, 3))
        self.assertEqual(app.get_preview_settings(None), (False, 384, 15, 3))

	import pickle
	import unittest

	from tests.test_glb_cache import app, gr


	class PreviewRequestHeaders(unittest.TestCase):
	def test_mobile_resolution_survives_request_serialization(self):
	for header in ("user-agent", "User-Agent"):
	request = gr.Request(headers={header: "Mozilla/5.0 (iPhone) Mobile"})
	restored = pickle.loads(pickle.dumps(request))
	self.assertEqual(app.get_preview_settings(restored), (True, 256, 15, 3))

	def test_desktop_and_missing_requests_keep_desktop_resolution(self):
	request = gr.Request(headers={"user-agent": "Mozilla/5.0 (Macintosh)"})
	self.assertEqual(app.get_preview_settings(request), (False, 384, 15, 3))
	self.assertEqual(app.get_preview_settings(None), (False, 384, 15, 3))