• Example Search
• Project Search

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• frameless-master
  • ml
    • src
      • main
        • scala
          • frameless
            • ml
              • internals
                • UnaryInputsChecker.scala
                • VectorInputsChecker.scala
                • SelectorByValue.scala
                • LinearInputsChecker.scala
                • TreesInputsChecker.scala
              • package.scala
              • classification
                • TypedRandomForestClassifier.scala
              • TypedEstimator.scala
              • clustering
                • TypedKMeans.scala
                • TypedBisectingKMeans.scala
              • params
                • trees
                  • FeatureSubsetStrategy.scala
                • linears
                  • LossStrategy.scala
                  • Solver.scala
                • kmeans
                  • KMeansInitMode.scala
              • regression
                • TypedRandomForestRegressor.scala
                • TypedLinearRegression.scala
              • feature
                • TypedVectorAssembler.scala
                • TypedIndexToString.scala
                • TypedStringIndexer.scala
              • TypedTransformer.scala
          • org
            • apache
              • spark
                • ml
                  • FramelessInternals.scala
      • test
        • scala
          • frameless
            • ml
              • FramelessMlSuite.scala
              • TypedEncoderInstancesTests.scala
              • Generators.scala
              • classification
                • ClassificationIntegrationTests.scala
                • TypedRandomForestClassifierTests.scala
              • clustering
                • BisectingKMeansTests.scala
                • ClusteringIntegrationTests.scala
                • KMeansTests.scala
              • regression
                • RegressionIntegrationTests.scala
                • TypedLinearRegressionTests.scala
                • TypedRandomForestRegressorTests.scala
              • feature
                • TypedVectorAssemblerTests.scala
                • TypedIndexToStringTests.scala
                • TypedStringIndexerTests.scala
  • dataset
    • src
      • main
        • scala
          • frameless
            • functions
              • UnaryFunctions.scala
              • AggregateFunctions.scala
              • package.scala
              • Udf.scala
              • NonAggregateFunctions.scala
              • Lit.scala
            • TypedEncoder.scala
            • Job.scala
            • FramelessSyntax.scala
            • syntax
              • package.scala
            • RecordEncoder.scala
            • TypedDatasetForwarded.scala
            • TypedDataset.scala
            • TypedColumn.scala
            • ops
              • Repeat.scala
              • AggregateTypes.scala
              • As.scala
              • ColumnTypes.scala
              • RelationalGroupsOps.scala
              • GroupByOps.scala
              • SmartProject.scala
            • TypedExpressionEncoder.scala
            • SparkDelay.scala
          • org
            • apache
              • spark
                • sql
                  • FramelessInternals.scala
      • test
        • resources
          • log4j.properties
        • scala
          • frameless
            • LitTests.scala
            • functions
              • UdfTests.scala
              • DateTimeStringBehaviourUtils.scala
              • UnaryFunctionsTest.scala
              • AggregateFunctionsTests.scala
              • DoubleBehaviourUtils.scala
              • NonAggregateFunctionsTests.scala
            • JobTests.scala
            • SelectTests.scala
            • TypedDatasetSuite.scala
            • package.scala
            • DropTest.scala
            • CastTests.scala
            • AsTests.scala
            • InjectionTests.scala
            • FilterTests.scala
            • RecordEncoderTests.scala
            • syntax
              • FramelessSyntaxTests.scala
            • CreateTests.scala
            • ColTests.scala
            • ExplodeTests.scala
            • NumericTests.scala
            • WithColumnTest.scala
            • CollectTests.scala
            • GroupByTests.scala
            • forward
              • FirstTests.scala
              • ToJSONTests.scala
              • LimitTests.scala
              • IsLocalTests.scala
              • ToLocalIteratorTests.scala
              • DistinctTests.scala
              • ForeachTests.scala
              • TakeTests.scala
              • WriteTests.scala
              • RandomSplitTests.scala
              • UnionTests.scala
              • CountTests.scala
              • CheckpointTests.scala
              • InputFilesTests.scala
              • HeadTests.scala
              • ExceptTests.scala
              • SparkSessionTests.scala
              • IntersectTests.scala
              • ColumnsTests.scala
              • QueryExecutionTests.scala
              • IsStreamingTests.scala
              • SQLContextTests.scala
              • StorageLevelTests.scala
            • ColumnTests.scala
            • FlattenTests.scala
            • WithColumnTupledTest.scala
            • XN.scala
            • BitwiseTests.scala
            • EncoderTests.scala
            • UdtEncodedClass.scala
            • ops
              • PivotTest.scala
              • RepeatTest.scala
              • deserialized
                • ReduceTests.scala
                • MapPartitionsTests.scala
                • FilterTests.scala
                • MapTests.scala
                • FlatMapTests.scala
              • CubeTests.scala
              • SmartProjectTest.scala
              • ColumnTypesTest.scala
              • RollupTests.scala
            • DropTupledTest.scala
            • SchemaTests.scala
            • JoinTests.scala
            • SelfJoinTests.scala
            • OrderByTests.scala
  • LICENSE
  • project
    • build.properties
    • plugins.sbt
  • cats
    • src
      • main
        • scala
          • frameless
            • cats
              • implicits.scala
              • SparkTask.scala
              • package.scala
              • SparkDelayInstances.scala
              • FramelessSyntax.scala
      • test
        • resources
          • log4j.properties
        • scala
          • frameless
            • cats
              • FramelessSyntaxTests.scala
              • test.scala
  • .travis.yml
  • README.md
  • scripts
    • docs-build.sh
    • travis-publish.sh
    • docs-publish.sh
  • core
    • src
      • main
        • scala
          • frameless
            • CatalystPivotable.scala
            • CatalystNotNullable.scala
            • CatalystOrdered.scala
            • CatalystBitShift.scala
            • CatalystCollection.scala
            • CatalystNumericWithJavaBigDecimal.scala
            • SQLTimestamp.scala
            • CatalystIsin.scala
            • CatalystDivisible.scala
            • CatalystRound.scala
            • CatalystNumeric.scala
            • CatalystAverageable.scala
            • Injection.scala
            • CatalystVariance.scala
            • SQLDate.scala
            • CatalystBitwise.scala
            • CatalystSummable.scala
            • CatalystCast.scala
            • CatalystNaN.scala
  • build.sbt
  • .gitignore
  • docs
    • src
      • main
        • tut
          • TypedEncoder.md
          • FeatureOverview.md
          • SUMMARY.md
          • TypedDatasetVsSparkDataset.md
          • Injection.md
          • Job.md
          • TypedML.md
          • Cats.md
          • TypedDataFrame.md
  • version.sbt

package frameless

import org.scalacheck.Prop
import org.scalacheck.Prop._
import scala.collection.JavaConverters._
import org.scalatest.matchers.should.Matchers

class ToLocalIteratorTests extends TypedDatasetSuite with Matchers {
  test("toLocalIterator") {
    def prop[A: TypedEncoder](data: Vector[A]): Prop = {
      val dataset = TypedDataset.create(data)

      dataset.toLocalIterator().run().asScala.toIterator sameElements dataset.dataset.toLocalIterator().asScala.toIterator
    }

    check(forAll(prop[Int] _))
    check(forAll(prop[String] _))
  }
}