fz0x1/Stegano
1
0
Fork 0
mirror of https://github.com/cedricbonhomme/Stegano.git synced 2025-06-28 03:06:14 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix: [core] various minor bug fixes.

Browse source
This commit is contained in:
Cédric Bonhomme 2022-01-05 00:29:13 +01:00
parent 0a17b29121
commit 1b8d8042b7
No known key found for this signature in database
GPG key ID: A1CB94DE57B7A70D
25 changed files with 122 additions and 148 deletions
Download patch file Download diff file Expand all files Collapse all files

10
tests/test_generators.py
View file

@ -2,7 +2,7 @@
# -*- coding: utf-8 -*-
# Stegano - Stegano is a pure Python steganography module.
# Copyright (C) 2010-2021 Cédric Bonhomme - https://www.cedricbonhomme.org
# Copyright (C) 2010-2022 Cédric Bonhomme - https://www.cedricbonhomme.org
#
# For more information : https://git.sr.ht/~cedric/stegano
#
@ -183,7 +183,7 @@ class TestGenerators(unittest.TestCase):
@staticmethod
def shi_tomashi_reconfigure(
file_name: str,
corners: int = 1000,
max_corners: int = 1000,
quality: float = 0.001,
min_distance: int = 10,
):
@ -192,8 +192,10 @@ class TestGenerators(unittest.TestCase):
"""
image = cv2.imread(file_name)
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
corners = cv2.goodFeaturesToTrack(gray, corners, quality, min_distance)
corners = np.int0(corners)
corners = cv2.goodFeaturesToTrack(gray, max_corners, quality, min_distance)
# Commented because min_distance argument of generators.shi_tomashi is now set
# to 10.0:
# corners = np.int0(corners)
corners = corners.reshape(corners.shape[0], -1)
np.savetxt("tests/expected-results/shi_tomashi.txt", corners)