import os
import sys

import lit.formats
import lit.util

import lit.llvm

# Configuration file for the 'lit' test runner.
lit.llvm.initialize(lit_config, config)

# name: The name of this test suite.
config.name = 'TFLITEHUB'

config.test_format = lit.formats.ShTest()

# suffixes: A list of file extensions to treat as test files.
config.suffixes = ['.py']

# test_source_root: The root path where tests are located.
config.test_source_root = os.path.dirname(__file__)

#config.use_default_substitutions()
config.excludes = [
  'lit.cfg.py',
  'lit.site.cfg.py',
  'test_util.py',
  'manual_test.py',
]

config.substitutions.extend([
    ('%PYTHON', sys.executable),
])

config.environment['PYTHONPATH'] = ":".join(sys.path)

project_root = os.path.dirname(os.path.dirname(__file__))

# Enable features based on -D FEATURES=hugetest,vulkan
# syntax.
features_param = lit_config.params.get('FEATURES')
if features_param:
  config.available_features.update(features_param.split(','))

@@ -521,16 +521,43 @@
 des = %5B%0A
+  'imagenet_test_data.py',%0A
   'lit.c
@@ -592,17 +592,19 @@
 %0A  '
-test_util
+manual_test
 .py'
@@ -604,35 +604,57 @@
 est.py',%0A  '
-manual
+squad
 _test
+_data.py',%0A  'test_util
 .py',%0A%5D%0A%0Acon

Exclude test data utilities from lit.