#!/usr/bin/env python3
# Copyright 2010-2021 Google LLC
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
"""Simple unit tests for python/init.swig. Not exhaustive."""

import unittest
from ortools.init import pywrapinit

class PyWrapInit(unittest.TestCase):

    def test_logging
        pywrapinit.CppBridge.InitLogging('pywrapinit_test.py')
        pywrapinit.CppBridge.ShutdownLogging()

    def test_flags(self):
        print('test_version')
        cpp_flags = pywrapinit.CppFlags()
        cpp_flags.logtostderr = True
        cpp_flags.log_prefix = True
        cpp_flags.cp_model_dump_prefix = 'pywrapinit_test'
        cpp_flags.cp_model_dump_model = True
        cpp_flags.cp_model_dump_lns = True
        cpp_flags.cp_model_dump_response = True
        pywrapinit.CppBridge.SetFlags(cpp_flags)

    def test_version(self):
        print('test_version')
        version = pywrapinit.OrToolsVersion
        self.assertEqual(9, version.MajorNumber())
        self.assertEqual(1, version.MinorNumber())
        self.assertTrue(isnumeric(version.PathNumber()))
        major = version.MajorNumber()
        minor = version.MinorNumber()
        patch = version.PatchNumber()
        string = f'{major}.{minor}.{patch}'
        self.assertEqual(string, version.VersionString())

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

@@ -767,16 +767,53 @@
 _logging
+(self):%0A        print('test_logging')
 %0A       
@@ -954,39 +954,41 @@
     print('test_
-version
+cpp_flags
 ')%0A        cpp_f
@@ -1400,23 +1400,21 @@
         
-version
+major
  = pywra
@@ -1433,16 +1433,30 @@
 sVersion
+.MajorNumber()
 %0A       
@@ -1471,69 +1471,66 @@
 sert
-Equal(9, version.MajorNumber())%0A        self.assertEqual(1, v
+IsInstance(major, int)%0A        minor = pywrapinit.OrToolsV
 ersi
@@ -1545,17 +1545,16 @@
 Number()
-)
 %0A       
@@ -1569,24 +1569,66 @@
 sert
-True(isnumeric(v
+IsInstance(minor, int)%0A        patch = pywrapinit.OrToolsV
 ersi
@@ -1633,16 +1633,17 @@
 sion.Pat
+c
 hNumber(
@@ -1643,18 +1643,16 @@
 Number()
-))
 %0A       
@@ -1656,112 +1656,144 @@
     
-major = version.MajorNumber()%0A        minor = version.MinorNumber()%0A        patch = version.PatchNumber(
+self.assertIsInstance(patch, int)%0A        version = pywrapinit.OrToolsVersion.VersionString()%0A        self.assertIsInstance(version, str
 )%0A  
@@ -1863,39 +1863,23 @@
 ual(
-string, version.VersionS
+version, s
 tring
-()
 )%0A%0Ai

Fix pywraptest_init.py