Skip to content

Commit 76797cc

Browse files
author
agrandiere
committed
add real test
1 parent 9862764 commit 76797cc

File tree

1 file changed

+13
-2
lines changed

1 file changed

+13
-2
lines changed

tests.py

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,8 +2,19 @@
22
from sightengine.client import SightengineClient
33

44
class Tests(unittest.TestCase):
5-
def test_upper(self):
6-
self.assertEqual('foo'.upper(), 'FOO')
5+
def test_nudity(self):
6+
client = SightengineClient('test', 'test')
7+
checkNudity = client.check('nudity')
8+
9+
output = checkNudity.image('https://d3m9459r9kwism.cloudfront.net/img/examples/example5.jpg')
10+
11+
self.assertEqual('success', output['status'])
12+
13+
output2 = checkNudity.image('https://incorrectUrl.jpg')
14+
15+
self.assertEqual('failure', output2['status'])
16+
self.assertEqual('media_error', output2['error']['type'])
17+
718

819
if __name__ == '__main__':
920
unittest.main()

0 commit comments

Comments
 (0)