56 lines
1.6 KiB
Python
56 lines
1.6 KiB
Python
# ------------------------------------------------------------------
|
|
# Copyright (c) 2020 PyInstaller Development Team.
|
|
#
|
|
# This file is distributed under the terms of the GNU General Public
|
|
# License (version 2.0 or later).
|
|
#
|
|
# The full license is available in LICENSE.GPL.txt, distributed with
|
|
# this software.
|
|
#
|
|
# SPDX-License-Identifier: GPL-2.0-or-later
|
|
# ------------------------------------------------------------------
|
|
|
|
import pytest
|
|
|
|
from PyInstaller.utils.tests import importorskip
|
|
|
|
|
|
# Run the tests in onedir mode only
|
|
tensorflow_onedir_only = pytest.mark.parametrize('pyi_builder', ['onedir'], indirect=True)
|
|
|
|
|
|
@importorskip('tensorflow')
|
|
@tensorflow_onedir_only
|
|
def test_tensorflow(pyi_builder):
|
|
pyi_builder.test_source(
|
|
"""
|
|
from tensorflow import *
|
|
"""
|
|
)
|
|
|
|
|
|
# Test if tensorflow.keras imports properly result in tensorflow being collected.
|
|
# See https://github.com/pyinstaller/pyinstaller/discussions/6890
|
|
@importorskip('tensorflow')
|
|
@tensorflow_onedir_only
|
|
def test_tensorflow_keras_import(pyi_builder):
|
|
pyi_builder.test_source(
|
|
"""
|
|
from tensorflow.keras.models import Sequential
|
|
from tensorflow.keras.layers import Dense, LSTM, Dropout
|
|
from tensorflow.keras.optimizers import Adam
|
|
"""
|
|
)
|
|
|
|
|
|
@importorskip('tensorflow')
|
|
@tensorflow_onedir_only
|
|
def test_tensorflow_layer(pyi_builder):
|
|
pyi_builder.test_script('pyi_lib_tensorflow_layer.py')
|
|
|
|
|
|
@importorskip('tensorflow')
|
|
@tensorflow_onedir_only
|
|
def test_tensorflow_mnist(pyi_builder):
|
|
pyi_builder.test_script('pyi_lib_tensorflow_mnist.py')
|