mirror of
https://github.com/getredash/redash.git
synced 2025-12-25 01:03:20 -05:00
Add support for int/float values in guess_type (#3913)
This commit is contained in:
@@ -1,8 +1,7 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
from unittest import TestCase
|
||||
|
||||
from redash.query_runner import TYPE_DATETIME, TYPE_FLOAT, TYPE_INTEGER, TYPE_BOOLEAN, TYPE_STRING
|
||||
from redash.query_runner.drill import guess_type
|
||||
from redash.query_runner import TYPE_DATETIME, TYPE_FLOAT, TYPE_INTEGER, TYPE_BOOLEAN, TYPE_STRING, guess_type
|
||||
|
||||
|
||||
class TestGuessType(TestCase):
|
||||
@@ -16,6 +15,7 @@ class TestGuessType(TestCase):
|
||||
self.assertEqual(guess_type('false'), TYPE_BOOLEAN)
|
||||
self.assertEqual(guess_type('False'), TYPE_BOOLEAN)
|
||||
self.assertEqual(guess_type('FALSE'), TYPE_BOOLEAN)
|
||||
self.assertEqual(guess_type(False), TYPE_BOOLEAN)
|
||||
|
||||
def test_detects_strings(self):
|
||||
self.assertEqual(guess_type(None), TYPE_STRING)
|
||||
@@ -24,9 +24,11 @@ class TestGuessType(TestCase):
|
||||
|
||||
def test_detects_integer(self):
|
||||
self.assertEqual(guess_type('42'), TYPE_INTEGER)
|
||||
self.assertEqual(guess_type(42), TYPE_INTEGER)
|
||||
|
||||
def test_detects_float(self):
|
||||
self.assertEqual(guess_type('3.14'), TYPE_FLOAT)
|
||||
self.assertEqual(guess_type(3.14), TYPE_FLOAT)
|
||||
|
||||
def test_detects_date(self):
|
||||
self.assertEqual(guess_type('2018-10-31'), TYPE_DATETIME)
|
||||
|
||||
Reference in New Issue
Block a user