mirror of
https://github.com/kanzure/pdfparanoia.git
synced 2025-02-11 13:13:10 +01:00
28 lines
813 B
Python
28 lines
813 B
Python
|
# -*- coding: utf-8 -*-
|
||
|
|
|
||
|
|
import unittest
|
||
|
|
from pdfparanoia.eraser import remove_object_by_id
|
||
|
|
|
||
|
|
class EraserTestCase(unittest.TestCase):
|
||
|
|
def test_remove_object_by_id(self):
|
||
|
|
content = ""
|
||
|
|
output = remove_object_by_id(content, 1)
|
||
|
|
self.assertEqual(content, output)
|
||
|
|
|
||
|
|
content = ""
|
||
|
|
output = remove_object_by_id(content, 2)
|
||
|
|
self.assertEqual(content, output)
|
||
|
|
|
||
|
|
content = ""
|
||
|
|
output = remove_object_by_id(content, 100)
|
||
|
|
self.assertEqual(content, output)
|
||
|
|
|
||
|
|
content = "1 0 obj\nthings\nendobj\nleftovers"
|
||
|
|
output = remove_object_by_id(content, 2)
|
||
|
|
self.assertEqual(content, output)
|
||
|
|
|
||
|
|
content = "1 0 obj\nthings\nendobj\nleftovers"
|
||
|
|
output = remove_object_by_id(content, 1)
|
||
|
|
self.assertEqual("leftovers", output)
|
||
|
|
|