79 lines
2.1 KiB
Python
79 lines
2.1 KiB
Python
#!/usr/bin/env python
|
|
from api_client import APIconnection
|
|
|
|
import os
|
|
import io
|
|
import sys
|
|
import time
|
|
import numpy as np
|
|
import uuid
|
|
|
|
from PIL import Image
|
|
|
|
import unittest
|
|
from pprint import pprint
|
|
|
|
import logging, sys
|
|
logging.basicConfig(stream=sys.stderr, level=logging.INFO)
|
|
|
|
|
|
class TestCapture(unittest.TestCase):
|
|
|
|
def test_capture_config(self):
|
|
connection = APIconnection(host="localhost", port=5000, api_ver="v1")
|
|
config = connection.get_config()
|
|
|
|
expected_keys = [
|
|
'image_resolution',
|
|
'video_resolution',
|
|
'numpy_resolution',
|
|
]
|
|
|
|
for key in expected_keys:
|
|
self.assertTrue(key in config)
|
|
|
|
def test_capture_videoport(self):
|
|
connection = APIconnection(host="localhost", port=5000, api_ver="v1")
|
|
resolution = connection.get_config()['video_resolution']
|
|
|
|
capture_array = connection.capture(
|
|
use_video_port=True,
|
|
keep_on_disk=False,
|
|
delete_after_use=True)
|
|
|
|
self.assertTrue(capture_array.shape == (resolution[1], resolution[0], 3))
|
|
|
|
def test_capture_full(self):
|
|
connection = APIconnection(host="localhost", port=5000, api_ver="v1")
|
|
resolution = connection.get_config()['image_resolution']
|
|
|
|
capture_array = connection.capture(
|
|
use_video_port=False,
|
|
keep_on_disk=False,
|
|
delete_after_use=True)
|
|
|
|
self.assertTrue(capture_array.shape == (resolution[1], resolution[0], 3))
|
|
|
|
class TestStage(unittest.TestCase):
|
|
|
|
def test_stage_config(self):
|
|
connection = APIconnection(host="localhost", port=5000, api_ver="v1")
|
|
config = connection.get_config()
|
|
|
|
expected_keys = [
|
|
'backlash',
|
|
]
|
|
|
|
for key in expected_keys:
|
|
self.assertTrue(key in config)
|
|
|
|
if __name__ == '__main__':
|
|
|
|
suites = [
|
|
unittest.TestLoader().loadTestsFromTestCase(TestCapture),
|
|
unittest.TestLoader().loadTestsFromTestCase(TestStage),
|
|
]
|
|
|
|
alltests = unittest.TestSuite(suites)
|
|
|
|
result = unittest.TextTestRunner(verbosity=2).run(alltests)
|