From a2c786fe64cb3f613efde2e3e1dbafec155ec6e4 Mon Sep 17 00:00:00 2001
From: David Hammer <dhammer@mailbox.org>
Date: Mon, 1 Nov 2021 16:00:19 +0100
Subject: [PATCH] Fix tests

---
 src/tests/test_agipd_kernels.py | 1 +
 src/tests/test_calcat_utils.py  | 3 +++
 2 files changed, 4 insertions(+)

diff --git a/src/tests/test_agipd_kernels.py b/src/tests/test_agipd_kernels.py
index 4a5aa10d..dd1fef68 100644
--- a/src/tests/test_agipd_kernels.py
+++ b/src/tests/test_agipd_kernels.py
@@ -37,6 +37,7 @@ kernel_runner = agipd_gpu.AgipdGpuRunner(
     constant_memory_cells=memory_cells,
     input_data_dtype=input_dtype,
     output_data_dtype=output_dtype,
+    output_gain_map=True,
 )
 
 
diff --git a/src/tests/test_calcat_utils.py b/src/tests/test_calcat_utils.py
index 537bddd9..69641d2c 100644
--- a/src/tests/test_calcat_utils.py
+++ b/src/tests/test_calcat_utils.py
@@ -70,6 +70,9 @@ class DummyDsscDevice:
     device_class_schema = Schema()
     managed_keys = []
 
+    def log_status_info(self, msg):
+        print(msg)
+
     @staticmethod
     def expectedParameters(expected):
         # super(DummyDsscDevice, DummyDsscDevice).expectedParameters(expected)
-- 
GitLab