FilesExpand file tree

 master

/

test_utils_predict.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

51 lines (39 loc) · 1.42 KB

 master

/

test_utils_predict.py

Copy path

Top

File metadata and controls

• Code
• Blame

51 lines (39 loc) · 1.42 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

import unittest

try:

from tests.unittests_helper import CustomTestCase

except ImportError:

from unittests_helper import CustomTestCase

import tensorflow as tf

import tensorlayer as tl

import numpy as np

class Util_Predict_Test(CustomTestCase):

@classmethod

def setUpClass(cls):

cls.x1 = tf.placeholder(tf.float32, [None, 5, 5, 3])

cls.x2 = tf.placeholder(tf.float32, [8, 5, 5, 3])

cls.X1 = np.ones([127, 5, 5, 3])

cls.X2 = np.ones([7, 5, 5, 3])

cls.batch_size = 8

@classmethod

def tearDownClass(cls):

tf.reset_default_graph()

def test_case1(self):

with self.assertNotRaises(Exception):

with tf.Session() as sess:

n = tl.layers.InputLayer(self.x1)

y = n.outputs

y_op = tf.nn.softmax(y)

tl.utils.predict(sess, n, self.X1, self.x1, y_op, batch_size=self.batch_size)

sess.close()

def test_case2(self):

with self.assertRaises(Exception):

with tf.Session() as sess:

n = tl.layers.InputLayer(self.x2)

y = n.outputs

y_op = tf.nn.softmax(y)

tl.utils.predict(sess, n, self.X2, self.x2, y_op, batch_size=self.batch_size)

sess.close()

if __name__ == '__main__':

# tf.logging.set_verbosity(tf.logging.INFO)

tf.logging.set_verbosity(tf.logging.DEBUG)

unittest.main()