regexp code cleanup

typelevel · Jun 1, 2023 · 6d1c69c · 6d1c69c
1 parent b9be1b5
commit 6d1c69c
Showing 1 changed file with 5 additions and 11 deletions.
diff --git a/dataset/src/test/scala/frameless/ColumnTests.scala b/dataset/src/test/scala/frameless/ColumnTests.scala
@@ -1,12 +1,14 @@
 package frameless
 
-import java.time.{Instant, Period, Duration}
+import java.time.{Duration, Instant, Period}
 import org.scalacheck.Prop._
-import org.scalacheck.{Arbitrary, Gen, Prop}, Arbitrary.arbitrary
+import org.scalacheck.{Arbitrary, Gen, Prop}
+import Arbitrary.arbitrary
 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 +206,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) =>