From c831b70040ce45772a9de0730fcdeef4adcedb7e Mon Sep 17 00:00:00 2001 From: Grigory Pomadchin Date: Thu, 1 Jun 2023 19:20:09 -0400 Subject: [PATCH] regexp code cleanup --- dataset/src/test/scala/frameless/ColumnTests.scala | 11 ++--------- 1 file changed, 2 insertions(+), 9 deletions(-) diff --git a/dataset/src/test/scala/frameless/ColumnTests.scala b/dataset/src/test/scala/frameless/ColumnTests.scala index e46dbc98..45ad2925 100644 --- a/dataset/src/test/scala/frameless/ColumnTests.scala +++ b/dataset/src/test/scala/frameless/ColumnTests.scala @@ -7,6 +7,7 @@ import org.scalatest.matchers.should.Matchers import shapeless.test.illTyped import scala.math.Ordering.Implicits._ +import scala.util.Try final class ColumnTests extends TypedDatasetSuite with Matchers { @@ -204,15 +205,7 @@ final class ColumnTests extends TypedDatasetSuite with Matchers { val spark = session import spark.implicits._ - val regex = - Gen.nonEmptyListOf(arbitrary[Char]).map(_.mkString).suchThat{ str => - try { - str.r - true - } catch { - case _: Throwable => false - } - } + val regex = Gen.nonEmptyListOf(arbitrary[Char]).map(_.mkString).suchThat { str => Try(str.r).isSuccess } check { forAll(regex, arbitrary[String]) { (a, b) =>