mocks.py 3.98 KB
Newer Older
André Anjos's avatar
André Anjos committed
1
2
3
#!/usr/bin/env python
# vim: set fileencoding=utf-8 :

Samuel GAIST's avatar
Samuel GAIST committed
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
###################################################################################
#                                                                                 #
# 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.            #
#                                                                                 #
###################################################################################
André Anjos's avatar
André Anjos committed
35
36
37
38
39


from ..data import DataSource
from ..data import DataSink

40

André Anjos's avatar
André Anjos committed
41
class MockDataSource(DataSource):
Philip ABBET's avatar
Philip ABBET committed
42
43
44
45
    def __init__(self, data, indexes):
        self.data = list(data)
        self.indexes = list(indexes)
        self.current = 0
André Anjos's avatar
André Anjos committed
46

Philip ABBET's avatar
Philip ABBET committed
47
    def next(self):
Samuel GAIST's avatar
Samuel GAIST committed
48
49
50
51
52
        result = (
            self.data[self.current],
            self.indexes[self.current][0],
            self.indexes[self.current][1],
        )
Philip ABBET's avatar
Philip ABBET committed
53
54
        self.current += 1
        return result
André Anjos's avatar
André Anjos committed
55

Philip ABBET's avatar
Philip ABBET committed
56
57
    def hasMoreData(self):
        return self.current < sum(1 for i in self.data)
André Anjos's avatar
André Anjos committed
58

59

Samuel GAIST's avatar
Samuel GAIST committed
60
# ----------------------------------------------------------
61
62


André Anjos's avatar
André Anjos committed
63
class MockDataSink(DataSink):
Philip ABBET's avatar
Philip ABBET committed
64
65
66
67
68
    class WrittenData:
        def __init__(self, data, start, end):
            self.data = data
            self.start = start
            self.end = end
André Anjos's avatar
André Anjos committed
69

Philip ABBET's avatar
Philip ABBET committed
70
71
72
73
    def __init__(self, dataformat):
        self.written = []
        self.can_write = True
        self.dataformat = dataformat
André Anjos's avatar
André Anjos committed
74

Philip ABBET's avatar
Philip ABBET committed
75
    def write(self, data, start_data_index, end_data_index):
Samuel GAIST's avatar
Samuel GAIST committed
76
77
        if not (self.can_write):
            raise IOError
Philip ABBET's avatar
Philip ABBET committed
78
        self.written.append(
Samuel GAIST's avatar
Samuel GAIST committed
79
80
            MockDataSink.WrittenData(data, start_data_index, end_data_index)
        )
André Anjos's avatar
André Anjos committed
81

Philip ABBET's avatar
Philip ABBET committed
82
83
    def isConnected(self):
        return True
84
85


Samuel GAIST's avatar
Samuel GAIST committed
86
# ----------------------------------------------------------
87
88
89


class MockDataSource_Crash(DataSource):
Philip ABBET's avatar
Philip ABBET committed
90
    def next(self):
Samuel GAIST's avatar
Samuel GAIST committed
91
        a = b  # noqa
92

Philip ABBET's avatar
Philip ABBET committed
93
    def hasMoreData(self):
Samuel GAIST's avatar
Samuel GAIST committed
94
        a = b  # noqa
Philip ABBET's avatar
Philip ABBET committed
95
        return False