コード例 #1
0
    def test_keep_streaming(self):
        dataset = Dataset(
                self.session,
                "Foo Name",
                title=self._TITLE,
                create=True,
                independents=self._INDEPENDENTS,
                dependents=self._DEPENDENTS)

        data = self._get_records_simple([(1, 2, 3)], dataset.data.dtype)

        dataset.addData(data)

        listener = 'listener'
        # Start streaming the listener
        dataset.keepStreaming(listener, 0)
        # Check the listener is notified of the data already available
        self.hub.onDataAvailable.assert_called_with(None, [listener])
        self.hub.reset_mock()
        # Keep streaming for more data added
        dataset.keepStreaming(listener, 1)
        # Add more data.
        dataset.addData(data)
        # Trigger the listener again.
        self.hub.onDataAvailable.assert_called_with(None, set([listener]))
コード例 #2
0
    def test_add_data_extended(self):
        dataset = Dataset(
                self.session,
                "Foo Name",
                title=self._TITLE,
                create=True,
                independents=self._EXT_INDEPENDENTS,
                dependents=self._EXT_DEPENDENTS,
                extended=True)

        dataset.listeners.add('foo listener')
        row_1 = (1, [[0, 1], [1, 0]], [[0, 1], [2, 3], [4, 5]])
        row_2 = (2, [[1, 0], [1, 0]], [[6, 7], [8, 3], [2, 1]])
        data = self._get_records_extended([row_1, row_2], dataset.data.dtype)

        # Add the data.
        dataset.addData(data)

        self.hub.onDataAvailable.assert_called_with(None, set(['foo listener']))

        data_in_dataset, count = dataset.getData(None, 0, simpleOnly=False)
        self.assertEqual(count, 2)
        self.assertArrayEqual(row_1[0], data_in_dataset[0][0])
        self.assertArrayEqual(row_1[1], data_in_dataset[0][1])
        self.assertArrayEqual(row_1[2], data_in_dataset[0][2])
        self.assertArrayEqual(row_2[0], data_in_dataset[1][0])
        self.assertArrayEqual(row_2[1], data_in_dataset[1][1])
        self.assertArrayEqual(row_2[2], data_in_dataset[1][2])
コード例 #3
0
    def test_save_reload_data_extended(self):
        # Set up the dataset
        dataset = Dataset(self.session,
                          "Foo Name",
                          title=self._TITLE,
                          create=True,
                          independents=self._EXT_INDEPENDENTS,
                          dependents=self._EXT_DEPENDENTS,
                          extended=True)

        dataset.listeners.add('foo listener')
        row_1 = (1, [[0, 1], [1, 0]], [[0, 1], [2, 3], [4, 5]])
        row_2 = (2, [[1, 0], [1, 0]], [[6, 7], [8, 3], [2, 1]])
        data = self._get_records_extended([row_1, row_2], dataset.data.dtype)
        # Add the data.
        dataset.addData(data)

        # Save the dataset
        dataset.save()

        # Create a new dataset that loads the data.
        new_dataset = Dataset(self.session,
                              "Foo Name",
                              title=self._TITLE,
                              extended=True)

        data_in_dataset, count = new_dataset.getData(None, 0, simpleOnly=False)
        self.assertEqual(count, 2)
        self.assertArrayEqual(row_1[0], data_in_dataset[0][0])
        self.assertArrayEqual(row_1[1], data_in_dataset[0][1])
        self.assertArrayEqual(row_1[2], data_in_dataset[0][2])
        self.assertArrayEqual(row_2[0], data_in_dataset[1][0])
        self.assertArrayEqual(row_2[1], data_in_dataset[1][1])
        self.assertArrayEqual(row_2[2], data_in_dataset[1][2])
コード例 #4
0
    def test_add_data_extended(self):
        dataset = Dataset(self.session,
                          "Foo Name",
                          title=self._TITLE,
                          create=True,
                          independents=self._EXT_INDEPENDENTS,
                          dependents=self._EXT_DEPENDENTS,
                          extended=True)

        dataset.listeners.add('foo listener')
        row_1 = (1, [[0, 1], [1, 0]], [[0, 1], [2, 3], [4, 5]])
        row_2 = (2, [[1, 0], [1, 0]], [[6, 7], [8, 3], [2, 1]])
        data = self._get_records_extended([row_1, row_2], dataset.data.dtype)

        # Add the data.
        dataset.addData(data)

        self.hub.onDataAvailable.assert_called_with(None,
                                                    set(['foo listener']))

        data_in_dataset, count = dataset.getData(None, 0, simpleOnly=False)
        self.assertEqual(count, 2)
        self.assertArrayEqual(row_1[0], data_in_dataset[0][0])
        self.assertArrayEqual(row_1[1], data_in_dataset[0][1])
        self.assertArrayEqual(row_1[2], data_in_dataset[0][2])
        self.assertArrayEqual(row_2[0], data_in_dataset[1][0])
        self.assertArrayEqual(row_2[1], data_in_dataset[1][1])
        self.assertArrayEqual(row_2[2], data_in_dataset[1][2])
コード例 #5
0
    def test_add_data_simple(self):
        dataset = Dataset(
                self.session,
                "Foo Name",
                title=self._TITLE,
                create=True,
                independents=self._INDEPENDENTS,
                dependents=self._DEPENDENTS)

        dataset.listeners.add('foo listener')

        data = self._get_records_simple(
                [(1, 2, 3), (2, 3, 4)], dataset.data.dtype)

        # Add the data.
        dataset.addData(data)

        self.hub.onDataAvailable.assert_called_with(None, set(['foo listener']))
        data_in_dataset, count = dataset.getData(None, 0, simpleOnly=True)
        self.assertEqual(count, 2)
        self.assertArrayEqual([1, 2, 3], data_in_dataset[0])
        self.assertArrayEqual([2, 3, 4], data_in_dataset[1])
コード例 #6
0
    def test_add_data_simple(self):
        dataset = Dataset(self.session,
                          "Foo Name",
                          title=self._TITLE,
                          create=True,
                          independents=self._INDEPENDENTS,
                          dependents=self._DEPENDENTS)

        dataset.listeners.add('foo listener')

        data = self._get_records_simple([(1, 2, 3), (2, 3, 4)],
                                        dataset.data.dtype)

        # Add the data.
        dataset.addData(data)

        self.hub.onDataAvailable.assert_called_with(None,
                                                    set(['foo listener']))
        data_in_dataset, count = dataset.getData(None, 0, simpleOnly=True)
        self.assertEqual(count, 2)
        self.assertArrayEqual([1, 2, 3], data_in_dataset[0])
        self.assertArrayEqual([2, 3, 4], data_in_dataset[1])
コード例 #7
0
    def test_save_reload_data_extended(self):
        # Set up the dataset
        dataset = Dataset(
                self.session,
                "Foo Name",
                title=self._TITLE,
                create=True,
                independents=self._EXT_INDEPENDENTS,
                dependents=self._EXT_DEPENDENTS,
                extended=True)

        dataset.listeners.add('foo listener')
        row_1 = (1, [[0, 1], [1, 0]], [[0, 1], [2, 3], [4, 5]])
        row_2 = (2, [[1, 0], [1, 0]], [[6, 7], [8, 3], [2, 1]])
        data = self._get_records_extended([row_1, row_2], dataset.data.dtype)
        # Add the data.
        dataset.addData(data)

        # Save the dataset
        dataset.save()

        # Create a new dataset that loads the data.
        new_dataset = Dataset(
                self.session,
                "Foo Name",
                title=self._TITLE,
                extended=True)

        data_in_dataset, count = new_dataset.getData(None, 0, simpleOnly=False)
        self.assertEqual(count, 2)
        self.assertArrayEqual(row_1[0], data_in_dataset[0][0])
        self.assertArrayEqual(row_1[1], data_in_dataset[0][1])
        self.assertArrayEqual(row_1[2], data_in_dataset[0][2])
        self.assertArrayEqual(row_2[0], data_in_dataset[1][0])
        self.assertArrayEqual(row_2[1], data_in_dataset[1][1])
        self.assertArrayEqual(row_2[2], data_in_dataset[1][2])
コード例 #8
0
    def test_keep_streaming(self):
        dataset = Dataset(self.session,
                          "Foo Name",
                          title=self._TITLE,
                          create=True,
                          independents=self._INDEPENDENTS,
                          dependents=self._DEPENDENTS)

        data = self._get_records_simple([(1, 2, 3)], dataset.data.dtype)

        dataset.addData(data)

        listener = 'listener'
        # Start streaming the listener
        dataset.keepStreaming(listener, 0)
        # Check the listener is notified of the data already available
        self.hub.onDataAvailable.assert_called_with(None, [listener])
        self.hub.reset_mock()
        # Keep streaming for more data added
        dataset.keepStreaming(listener, 1)
        # Add more data.
        dataset.addData(data)
        # Trigger the listener again.
        self.hub.onDataAvailable.assert_called_with(None, set([listener]))