Exemplos de StructType.colsOfType em Python

Exemplo n.º 1

Arquivo: test_struct_type.py Projeto: machielg/sparkle-session

 def test_str_type_names(self):
     s = StructType(fields=[
         StructField("a", IntegerType()),
         StructField("b", IntegerType())
     ])
     s = sparkle_struct_type(s)
     self.assertListEqual(['a', 'b'], s.colsOfType('int'))
     self.assertListEqual(['a', 'b'], s.colsOfType('integer'))

Exemplo n.º 2

Arquivo: test_struct_type.py Projeto: machielg/sparkle-session

 def test_array_type(self):
     s = StructType(fields=[StructField("a", ArrayType(StringType()))])
     s = sparkle_struct_type(s)
     self.assertListEqual(['a'], s.colsOfType('array'))
     self.assertListEqual(['a'], s.colsOfType('array<string>'))
     self.assertListEqual(['a'], s.colsOfType(ArrayType))
     self.assertListEqual(['a'], s.colsOfType(ArrayType(StringType())))
     self.assertListEqual([], s.colsOfType('int'))
     self.assertListEqual([], s.colsOfType('array<int>'))
     self.assertListEqual([], s.colsOfType(ArrayType(IntegerType())))