Beispiel #1
0
    def test_append(self):
        """Test the append function."""
        da = DoubleArray(0)
        da.append(1.0)
        da.append(2.0)
        da.append(3.0)

        self.assertEqual(da.length, 3)
        self.assertEqual(da[0], 1.0)
        self.assertEqual(da[1], 2.0)
        self.assertEqual(da[2], 3.0)
Beispiel #2
0
    def test_squeeze(self):
        """Tests the squeeze function."""
        da = DoubleArray(5)
        da.append(4.0)

        self.assertEqual(da.alloc > da.length, True)

        da.squeeze()

        self.assertEqual(da.length, 6)
        self.assertEqual(da.alloc == da.length, True)
        self.assertEqual(len(da.get_npy_array()), 6)