def test_read_three_lines(self):
reader = FileReader(
os.path.dirname(__file__) + "/data/file_3lines.txt")
self.assertEqual("abc", reader.read_line())
self.assertEqual("def", reader.read_line())
self.assertEqual("123", reader.read_line())
self.assertIsNone(reader.read_line())
def __init__(self, width=1280, height=720, parent=None):
QLabel.__init__(self, parent)
self.xpos = []
self.ypos = []
self.widths = []
self.heights = []
self.rect_centers = []
self.color_indices = []
self.recent_draw = [] # Flag to indicate if bbox was created now.
self.image_width = width
self.image_height = height
self.image = QPixmap(self.image_width, self.image_height)
self.original_size = self.image.size()
self.scaled_image = QPixmap()
self.original_scaled_image = self.scaled_image.copy()
self.scaled_image_to_draw = self.scaled_image.copy()
self.setFixedSize(self.scaled_image.size())
self.current_color_index = 0
self.update_pixmap(self.image)
self.drawing = False
self.last_point = None
self.file_reader = FileReader()
def __init__(self, parent=None, extension='.png'):
super().__init__(parent)
self.mode = utils.MODE_TRACK
self.tracker_name = 'default'
self.trackers = []
self.current_class = 'default'
self.current_color_index = 0
self.class_colors = {}
self.prev_index_frame = 0
self.current_index_frame = 0
self.folder_reader = FolderReader()
self.image_directory = './imagen'
self.extension = extension
self.file_saver = FileSaver(self.image_directory)
self.file_reader = FileReader()
# self.filenames = ['out2.png', 'out19.png']
self.filenames = utils.read_files(self.image_directory, self.extension)
self.filenames = utils.sort_files(self.filenames)
# Timer for run button
self.run_timer = QTimer()
