Coverage for tests/images/test_wrapped_image_server.py: 94%

1import numpy as np 

2import dask.array as da 

3from qubalab.images.image_server import ImageMetadata, ImageServer 

4from qubalab.images.wrapped_image_server import WrappedImageServer 

5from qubalab.images.metadata.image_shape import ImageShape 

6from qubalab.images.metadata.pixel_calibration import PixelCalibration, PixelLength 

7from qubalab.images.region_2d import Region2D 

8 

9 

10class SampleServer(ImageServer): 

11 isClosed = False 

12 

13 def close(self): 

14 self.isClosed = True 

15 

16 def _build_metadata(self) -> ImageMetadata: 

17 return ImageMetadata( 

18 "/path/to/img.tiff", 

19 "Image name", 

20 ( 

21 ImageShape(64, 50), 

22 ImageShape(32, 25), 

23 ImageShape(16, 12) 

24 ), 

25 PixelCalibration( 

26 PixelLength.create_microns(2.5), 

27 PixelLength.create_microns(2.5) 

28 ), 

29 True, 

30 np.uint8 

31 ) 

32 

33 def _read_block(self, level: int, region: Region2D) -> np.ndarray: 

34 return np.array() 

35 

36 

37class SampleWrappedServer(WrappedImageServer): 

38 def _read_block(self, level: int, region: Region2D) -> np.ndarray: 

39 return np.array() 

40 

41 

42def test_wrapped_server(): 

43 expected_sample_server = SampleServer() 

44 wrapped_server = SampleWrappedServer(expected_sample_server) 

45 

46 sample_server = wrapped_server.base_server 

47 

48 assert sample_server == expected_sample_server 

49 

50 wrapped_server.close() 

51 

52 

53def test_sample_server_closed(): 

54 sample_server = SampleServer() 

55 wrapped_server = SampleWrappedServer(sample_server) 

56 

57 wrapped_server.close() 

58 

59 assert sample_server.isClosed 

60 

61 

62def test_metadata(): 

63 sample_server = SampleServer() 

64 wrapped_server = SampleWrappedServer(sample_server) 

65 expected_metadata = sample_server.metadata 

66 

67 metadata = wrapped_server.metadata 

68 

69 assert expected_metadata == metadata 

70 

71 wrapped_server.close()