import tensorflow as tf

        net, end_points = mobilenet_v1.mobilenet_v1(inputs, num_classes)
        self.assertTrue(net.op.name.startswith('MobilenetV1/Logits/AvgPool'))
        self.assertListEqual(net.get_shape().as_list(), [batch_size, 1, 1, 1024])
        self.assertFalse('Logits' in end_points)
        self.assertFalse('Predictions' in end_points)

    def testBuildBaseNetwork(self):
        batch_size = 5
        height, width = 224, 224

        inputs = tf.random_uniform((batch_size, height, width, 3))
        net, end_points = mobilenet_v1.mobilenet_v1_base(inputs)
        self.assertTrue(net.op.name.startswith('MobilenetV1/Conv2d_13'))
        self.assertListEqual(net.get_shape().as_list(),
                             [batch_size, 7, 7, 1024])
        expected_endpoints = ['Conv2d_0',
                              'Conv2d_1_depthwise', 'Conv2d_1_pointwise',
                              'Conv2d_2_depthwise', 'Conv2d_2_pointwise',
                              'Conv2d_3_depthwise', 'Conv2d_3_pointwise',
                              'Conv2d_4_depthwise', 'Conv2d_4_pointwise',
