From 9e3a9c08b26856c8466d37c1e82ec805140b711f Mon Sep 17 00:00:00 2001
From: Samuel Gaist <samuel.gaist@idiap.ch>
Date: Thu, 18 Apr 2019 09:05:30 +0200
Subject: [PATCH] [test][prefix][database] Fix simple/2 code

The migration script didn't properly copy the code.
---
 beat/core/test/prefix/databases/simple/2.py | 97 +++++++++++++++++----
 1 file changed, 80 insertions(+), 17 deletions(-)

diff --git a/beat/core/test/prefix/databases/simple/2.py b/beat/core/test/prefix/databases/simple/2.py
index e306693b..17736c90 100644
--- a/beat/core/test/prefix/databases/simple/2.py
+++ b/beat/core/test/prefix/databases/simple/2.py
@@ -1,22 +1,85 @@
-class View:
-    def setup(
-        self,
-        root_folder,
-        outputs,
-        parameters,
-        force_start_index=None,
-        force_end_index=None,
-    ):
-        """Initializes the database"""
+#!/usr/bin/env python
+# vim: set fileencoding=utf-8 :
 
-        return True
+###################################################################################
+#                                                                                 #
+# Copyright (c) 2019 Idiap Research Institute, http://www.idiap.ch/               #
+# Contact: beat.support@idiap.ch                                                  #
+#                                                                                 #
+# Redistribution and use in source and binary forms, with or without              #
+# modification, are permitted provided that the following conditions are met:     #
+#                                                                                 #
+# 1. Redistributions of source code must retain the above copyright notice, this  #
+# list of conditions and the following disclaimer.                                #
+#                                                                                 #
+# 2. Redistributions in binary form must reproduce the above copyright notice,    #
+# this list of conditions and the following disclaimer in the documentation       #
+# and/or other materials provided with the distribution.                          #
+#                                                                                 #
+# 3. Neither the name of the copyright holder nor the names of its contributors   #
+# may be used to endorse or promote products derived from this software without   #
+# specific prior written permission.                                              #
+#                                                                                 #
+# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND #
+# ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED   #
+# WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE          #
+# DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE    #
+# FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL      #
+# DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR      #
+# SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER      #
+# CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY,   #
+# OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE   #
+# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.            #
+#                                                                                 #
+###################################################################################
 
-    def done(self):
-        """Should return ``True``, when data is finished"""
 
-        return True
+import numpy
 
-    def next(self):
-        """Loads the next data block on ``outputs``"""
+from collections import namedtuple
+from beat.backend.python.database import View as BaseView
 
-        return True
+
+class View(BaseView):
+    def index(self, root_folder, parameters):
+        Entry = namedtuple("Entry", ["out"])
+
+        return [Entry(42)]
+
+    def get(self, output, index):
+        obj = self.objs[index]
+
+        if output == "out":
+            return {"value": numpy.int32(obj.out)}
+
+
+# ----------------------------------------------------------
+
+
+class View2(BaseView):
+    def index(self, root_folder, parameters):
+        Entry = namedtuple("Entry", ["out"])
+
+        return [Entry(53)]
+
+    def get(self, output, index):
+        obj = self.objs[index]
+
+        if output == "out":
+            return {"value": numpy.int32(obj.out)}
+
+
+# ----------------------------------------------------------
+
+
+class LargeView(BaseView):
+    def index(self, root_folder, parameters):
+        Entry = namedtuple("Entry", ["out"])
+
+        return [Entry(0), Entry(1), Entry(2), Entry(3), Entry(4)]
+
+    def get(self, output, index):
+        obj = self.objs[index]
+
+        if output == "out":
+            return {"value": numpy.int32(obj.out)}
-- 
GitLab