regexp code cleanup

typelevel · Jun 1, 2023 · c831b70 · c831b70
1 parent b9be1b5
commit c831b70
Showing 1 changed file with 2 additions and 9 deletions.
diff --git 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) =>