функция, работающая с функциями Array [T] или List [T] или Iterable [T] - PullRequest
Новая
       35

функция, работающая с функциями Array [T] или List [T] или Iterable [T]

4 голосов
/ 12 февраля 2010

Я пытался написать функцию тестирования / хронометража для ответов, предоставленных в этом ТАК вопросе . Некоторые ответы работают на Array[T], некоторые на List[T], один на Iterable[T] и один на String!

Я хотел бы написать функцию, которая берет функции shift* из вопроса или ответов, список ввода, предикат и ожидаемый результат и запускает функцию. Вроде как: 

def test[T](
  func:(Seq[T], T=>Boolean) => Seq[T],
  input:Seq[T],
  predicate:T=>Boolean,
  expected:Seq[T]): Unit = {
    // may be some warm up
    // ... time start, run func, time stop, 
    // check output against expected
}

За исключением того, что я могу определить сигнатуру, поскольку Array, кажется, имеет изменяемые свойства Seq, в то время как List, кажется, имеет неизменные Seq свойства.

Какой лучший способ справиться с этим?

Редактировать : Используя предложение Томаса, я могу приблизиться (работает на Array[Char], List[T], но не на Array[T]): 

val inputArr = Array('a', 'b', 'C', 'D')
val expectArr = Array('a', 'C', 'D', 'b')
val inputList = inputArr.toList
val expectList = expectArr.toList

def test[I, T](
  func:(I, T=>Boolean) => Traversable[T],
  input: I,
  predicate: T=>Boolean,
  expected: Traversable[T]): Boolean = {
  val result = func(input, predicate)
  if (result.size == expected.size) {
    result.toIterable.zip(expected.toIterable).forall(x => x._1 == x._2)
  } else {
    false
  }
}

// this method is from Geoff [there][2] 
def shiftElements[A](l: List[A], pred: A => Boolean): List[A] = {
  def aux(lx: List[A], accum: List[A]): List[A] = {
    lx match {
      case Nil => accum
      case a::b::xs if pred(b) && !pred(a) => aux(a::xs, b::accum)
      case x::xs => aux(xs, x::accum)
    }
  }
  aux(l, Nil).reverse
}

def shiftWithFor[T](a: Array[T], p: T => Boolean):Array[T] = {
  for (i <- 0 until a.length - 1; if !p(a(i)) && p(a(i+1))) {
    val tmp = a(i); a(i) = a(i+1); a(i+1) = tmp
  }
  a
}

def shiftWithFor2(a: Array[Char], p: Char => Boolean):Array[Char] = {
  for (i <- 0 until a.length - 1; if !p(a(i)) && p(a(i+1))) {
    val tmp = a(i); a(i) = a(i+1); a(i+1) = tmp
  }
  a
}

def shiftMe_?(c:Char): Boolean = c.isUpper

println(test(shiftElements[Char], inputList, shiftMe_?, expectList))
println(test(shiftWithFor2, inputArr, shiftMe_?, expectArr))
//following line does not compile
println(test(shiftWithFor, inputArr, shiftMe_?, expectArr))
//found   : [T](Array[T], (T) => Boolean) => Array[T]
//required: (?, (?) => Boolean) => Traversable[?]

//following line does not compile
println(test(shiftWithFor[Char], inputArr, shiftMe_?, expectArr))
//found   : => (Array[Char], (Char) => Boolean) => Array[Char]
//required: (?, (?) => Boolean) => Traversable[?]

//following line does not compile
println(test[Array[Char], Char](shiftWithFor[Char], inputArr, shiftMe_?, expectArr))
//found   : => (Array[Char], (Char) => Boolean) => Array[Char]
//required: (Array[Char], (Char) => Boolean) => Traversable[Char]

Я отмечу ответ Даниэля как принятый, так как он компилируется и предоставляет мне другой способ достижения того, чего я хотел - если только метод Array [T] не создает новый массив (и не вызывает проблем с Manifest).

(2): Каким был бы функциональный подход к смещению определенных элементов массива?

Ответы [ 3 ]

3 голосов
/ 12 февраля 2010

Одним из способов является определение функции: 

def test[I, T](
  func:(I, T=>Boolean) => Traversable[T],
  input: I,
  predicate: T=>Boolean,
  expected: Traversable[T]): Unit = {
  println(func(input, predicate))
}

def g(x : Char) = true

test((x : String, y: Char => Boolean) => x, "asdf", g _ , "expected")
test((x : List[Char], y: Char => Boolean) => x, List('s'), g _, List('e'))
test((x : Array[Char], y: Char => Boolean) => x, Array('s'), g _, Array('e'))
test((x : Iterable[Char], y: Char => Boolean) => x, Set('s'), g _, Set('e'))
1 голос
/ 12 февраля 2010

Я бы использовал scala.collection.Seq в Scala 2.8, так как этот тип является родительским для всех упорядоченных коллекций. За исключением Array и String, к сожалению. Обойти это можно с помощью границ вида, например: 

def test
  [A, CC <% scala.collection.Seq[A]]
  (input: CC, expected: CC)
  (func: (CC, A => Boolean) => CC, predicate: A => Boolean): Unit = {
  def times(n: Int)(f: => Unit) = 1 to n foreach { count => f }
  def testFunction = assert(func(input, predicate) == expected)
  def warm = times(50) { testFunction }
  def test = times(50) { testFunction }

  warm
  val start = System.currentTimeMillis()
  test
  val end = System.currentTimeMillis()
  println("Total time "+(end - start))
}

Я каррирую эту функцию, чтобы ввод (и ожидаемый) можно было использовать для вывода типа. В любом случае, это не будет работать с вашей собственной версией Array в Scala 2.8, потому что для этого требуется Manifest. Я уверен, что это может быть предоставлено здесь как-то, но я не совсем понимаю, как.

Но скажем, вы игнорируете все эти вещи о последовательностях, массивах и т. Д. Просто удалите представление, связанное с функцией, и вы получите следующее: 

def test
  [A, CC]
  (input: CC, expected: CC)
  (func: (CC, A => Boolean) => CC, predicate: A => Boolean): Unit = {
  def times(n: Int)(f: => Unit) = 1 to n foreach { count => f }
  def testFunction = assert(func(input, predicate) == expected)
  def warm = times(50) { testFunction }
  def test = times(50) { testFunction }

  warm
  val start = System.currentTimeMillis()
  test
  val end = System.currentTimeMillis()
  println("Total time "+(end - start))
}

Который будет работать так же, как найти. Пока совпадение типов, для этой программы не имеет значения, что такое CC.

1 голос
/ 12 февраля 2010

Все типы, которые вы упоминаете (даже String), eithr явно (List) или неявно (Array and String) Iterable, поэтому все, что вам нужно сделать, это использовать Iterable в сигнатуре вашего метода, где вы теперь используете Seq.

...