|
3 | 3 |
|
4 | 4 | import ndx_ophys_devices |
5 | 5 | import numpy as np |
6 | | -import pynwb.base |
7 | | -from ndx_ophys_devices import ExcitationSource, OpticalFilter, Photodetector, DichroicMirror, Indicator |
| 6 | +from ndx_ophys_devices import ExcitationSource, OpticalFilter, Photodetector, DichroicMirror |
8 | 7 |
|
9 | 8 | from ndx_ophys_devices.testing import ( |
10 | 9 | mock_ExcitationSource, |
|
14 | 13 | ) |
15 | 14 |
|
16 | 15 | from pynwb.testing.mock.utils import name_generator |
17 | | - |
| 16 | +from pynwb.core import DynamicTableRegion |
18 | 17 | import ndx_microscopy |
19 | 18 |
|
20 | 19 |
|
@@ -396,31 +395,58 @@ def mock_PlanarMicroscopySeries( |
396 | 395 | return planar_microscopy_series |
397 | 396 |
|
398 | 397 |
|
| 398 | +def mock_PlanarMicroscopyStaticImage( |
| 399 | + *, |
| 400 | + microscopy_rig: ndx_microscopy.MicroscopyRig, |
| 401 | + planar_imaging_space: ndx_microscopy.PlanarImagingSpace, |
| 402 | + microscopy_channel: ndx_microscopy.MicroscopyChannel, |
| 403 | + name: Optional[str] = None, |
| 404 | + description: str = "A mock instance of a PlanarMicroscopyStaticImage type to be used for rapid testing.", |
| 405 | + data: Optional[np.ndarray] = None, |
| 406 | +) -> ndx_microscopy.PlanarMicroscopyStaticImage: |
| 407 | + image_name = name or name_generator("PlanarMicroscopyStaticImage") |
| 408 | + image_data = data if data is not None else np.ones(shape=(5, 5)) |
| 409 | + |
| 410 | + planar_microscopy_static_image = ndx_microscopy.PlanarMicroscopyStaticImage( |
| 411 | + name=image_name, |
| 412 | + description=description, |
| 413 | + microscopy_rig=microscopy_rig, |
| 414 | + microscopy_channel=microscopy_channel, |
| 415 | + planar_imaging_space=planar_imaging_space, |
| 416 | + data=image_data, |
| 417 | + ) |
| 418 | + return planar_microscopy_static_image |
| 419 | + |
| 420 | + |
399 | 421 | def mock_MultiPlaneMicroscopyContainer( |
400 | 422 | *, |
401 | | - planar_microscopy_series: List[ndx_microscopy.PlanarMicroscopySeries], |
| 423 | + planar_microscopy_static_images: Optional[List[ndx_microscopy.PlanarMicroscopyStaticImage]] = None, |
| 424 | + planar_microscopy_series: Optional[List[ndx_microscopy.PlanarMicroscopySeries]] = None, |
402 | 425 | name: Optional[str] = None, |
403 | 426 | ) -> ndx_microscopy.MultiPlaneMicroscopyContainer: |
404 | 427 | container_name = name or name_generator("MultiPlaneMicroscopyContainer") |
405 | | - |
406 | 428 | multi_plane_microscopy_container = ndx_microscopy.MultiPlaneMicroscopyContainer( |
407 | | - name=container_name, planar_microscopy_series=planar_microscopy_series |
| 429 | + name=container_name, |
| 430 | + planar_microscopy_static_images=planar_microscopy_static_images, |
| 431 | + planar_microscopy_series=planar_microscopy_series, |
408 | 432 | ) |
409 | 433 |
|
410 | 434 | return multi_plane_microscopy_container |
411 | 435 |
|
412 | 436 |
|
413 | 437 | def mock_MultiChannelMicroscopyContainer( |
414 | 438 | *, |
415 | | - microscopy_series: List[ndx_microscopy.MicroscopySeries], |
| 439 | + microscopy_static_images: Optional[List[ndx_microscopy.MicroscopyStaticImage]] = None, |
| 440 | + microscopy_series: Optional[List[ndx_microscopy.MicroscopySeries]] = None, |
416 | 441 | name: Optional[str] = None, |
417 | 442 | ) -> ndx_microscopy.MultiChannelMicroscopyContainer: |
418 | 443 | container_name = name or name_generator("MultiChannelMicroscopyContainer") |
419 | 444 |
|
420 | 445 | multi_channel_microscopy_container = ndx_microscopy.MultiChannelMicroscopyContainer( |
421 | | - name=container_name, microscopy_series=microscopy_series |
| 446 | + name=container_name, |
| 447 | + microscopy_static_images=microscopy_static_images, |
| 448 | + microscopy_series=microscopy_series, |
422 | 449 | ) |
423 | | - |
424 | 450 | return multi_channel_microscopy_container |
425 | 451 |
|
426 | 452 |
|
@@ -477,9 +503,32 @@ def mock_VolumetricMicroscopySeries( |
477 | 503 | return volumetric_microscopy_series |
478 | 504 |
|
479 | 505 |
|
| 506 | +def mock_VolumetricMicroscopyStaticImage( |
| 507 | + *, |
| 508 | + microscopy_rig: ndx_microscopy.MicroscopyRig, |
| 509 | + volumetric_imaging_space: ndx_microscopy.VolumetricImagingSpace, |
| 510 | + microscopy_channel: ndx_microscopy.MicroscopyChannel, |
| 511 | + name: Optional[str] = None, |
| 512 | + description: str = "A mock instance of a VolumetricMicroscopyStaticImage type to be used for rapid testing.", |
| 513 | + data: Optional[np.ndarray] = None, |
| 514 | +) -> ndx_microscopy.VolumetricMicroscopyStaticImage: |
| 515 | + image_name = name or name_generator("VolumetricMicroscopyStaticImage") |
| 516 | + image_data = data if data is not None else np.ones(shape=(5, 5, 5)) |
| 517 | + |
| 518 | + volumetric_microscopy_static_image = ndx_microscopy.VolumetricMicroscopyStaticImage( |
| 519 | + name=image_name, |
| 520 | + description=description, |
| 521 | + microscopy_rig=microscopy_rig, |
| 522 | + microscopy_channel=microscopy_channel, |
| 523 | + volumetric_imaging_space=volumetric_imaging_space, |
| 524 | + data=image_data, |
| 525 | + ) |
| 526 | + return volumetric_microscopy_static_image |
| 527 | + |
| 528 | + |
480 | 529 | def mock_MicroscopyResponseSeries( |
481 | 530 | *, |
482 | | - rois: pynwb.core.DynamicTableRegion, |
| 531 | + rois: DynamicTableRegion, |
483 | 532 | name: Optional[str] = None, |
484 | 533 | description: str = "A mock instance of a MicroscopyResponseSeries type to be used for rapid testing.", |
485 | 534 | data: Optional[np.ndarray] = None, |
|
0 commit comments