Ejemplo n.º 1
0
    def test_append(self):
        """Test the append function."""
        lla = LongLongArray(0)
        lla.append(1)
        lla.append(2)
        lla.append(3)

        self.assertEqual(lla.length, 3)
        self.assertEqual(lla[0], 1)
        self.assertEqual(lla[1], 2)
        self.assertEqual(lla[2], 3)
Ejemplo n.º 2
0
    def test_squeeze(self):
        """Tests the squeeze function."""
        lla = LongLongArray(5)
        lla.append(4)

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

        lla.squeeze()

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