fix: column values with NaN (#26946)

(cherry picked from commit d8a9847503)
This commit is contained in:
Beto Dealmeida
2024-02-02 12:56:02 -05:00
committed by Michael S. Molina
parent d1dad1e63d
commit 383b4d949f
4 changed files with 35 additions and 12 deletions

View File

@@ -72,6 +72,16 @@ class TestDatasourceApi(SupersetTestCase):
response = json.loads(rv.data.decode("utf-8"))
self.assertEqual(response["result"], [None])
@pytest.mark.usefixtures("app_context", "virtual_dataset")
def test_get_column_values_integers_with_nulls(self):
self.login(username="admin")
table = self.get_virtual_dataset()
rv = self.client.get(f"api/v1/datasource/table/{table.id}/column/col6/values/")
self.assertEqual(rv.status_code, 200)
response = json.loads(rv.data.decode("utf-8"))
for val in [1, None, 3, 4, 5, 6, 7, 8, 9, 10]:
assert val in response["result"]
@pytest.mark.usefixtures("app_context", "virtual_dataset")
def test_get_column_values_invalid_datasource_type(self):
self.login(username="admin")