diff --git a/codegen/smithy-dafny-codegen-test/src/test/java/software/amazon/polymorph/smithypython/PythonTestModels.java b/codegen/smithy-dafny-codegen-test/src/test/java/software/amazon/polymorph/smithypython/PythonTestModels.java index dd9190e5d..1632f9591 100644 --- a/codegen/smithy-dafny-codegen-test/src/test/java/software/amazon/polymorph/smithypython/PythonTestModels.java +++ b/codegen/smithy-dafny-codegen-test/src/test/java/software/amazon/polymorph/smithypython/PythonTestModels.java @@ -31,7 +31,6 @@ class PythonTestModels extends TestModelTest { DISABLED_TESTS.add("SimpleTypes/SimpleFloat"); DISABLED_TESTS.add("SimpleTypes/SimpleShort"); DISABLED_TESTS.add("SimpleTypes/SimpleTimestamp"); - DISABLED_TESTS.add("Streaming"); DISABLED_TESTS.add("SQSExtended"); DISABLED_TESTS.add("aws-sdks/ddb-lite"); DISABLED_TESTS.add("aws-sdks/glue"); diff --git a/codegen/smithy-dafny-codegen-test/src/test/java/software/amazon/polymorph/smithyrust/RustTestModels.java b/codegen/smithy-dafny-codegen-test/src/test/java/software/amazon/polymorph/smithyrust/RustTestModels.java index e1143e409..7155316fb 100644 --- a/codegen/smithy-dafny-codegen-test/src/test/java/software/amazon/polymorph/smithyrust/RustTestModels.java +++ b/codegen/smithy-dafny-codegen-test/src/test/java/software/amazon/polymorph/smithyrust/RustTestModels.java @@ -48,7 +48,9 @@ class RustTestModels extends TestModelTest { @ParameterizedTest @MethodSource("discoverTestModels") - void testModelsForRust(String relativeTestModelPath) { + protected void testModels(String relativeTestModelPath) { + super.testModels(relativeTestModelPath); + Assumptions.assumeFalse(DISABLED_TESTS.contains(relativeTestModelPath)); Path testModelPath = getTestModelPath(relativeTestModelPath);