def test_keras_import(self): model = Sequential() model.add(RepeatVector(3, input_shape=(10,))) model.build() self.keras_type_test(model, 0, 'RepeatVector')