Пример #1
0
    def test_numpy_parameter(self):
        name = 'dummy'
        data = np.array([1.1, 1.2, 1.3, 1.4])
        alignment = np.array([0.01, 0.02, 0.03, 0.04])
        misalignment = np.array([0.01, 0.02, 0.03])

        with self.assertRaises(TypeError):
            pd = NumpyParameterData(1, data, alignment)
        with self.assertRaises(TypeError):
            pd = NumpyParameterData(name, 'hi', alignment)
        with self.assertRaises(TypeError):
            pd = NumpyParameterData(name, data, 'hi')
        with self.assertRaises(ValueError):
            pd = NumpyParameterData(name, data, misalignment)

        pd = NumpyParameterData(name, data, alignment)

        self.assertEqual(id(data), id(pd.get_data_as_numpy_array(pd.get_alignment())))

        self.assertEqual(0, pd.get_data_as_numpy_array(np.array([0.05, 0.015])).size)
        self.assertEqual(np.array([1.2, 1.4]).all(), pd.get_data_as_numpy_array(np.array([0.04, 0.02])).all())