Unverified Commit 709d3728 authored by nabokovas's avatar nabokovas Committed by GitHub
Browse files

[tests][python] remove excess iterations (#3504)

parent f4b9519b
......@@ -215,7 +215,6 @@ class TestBasic(unittest.TestCase):
# test that method works for different data types
d1 = lgb.Dataset(x_1, feature_name=names, free_raw_data=False).construct()
res_feature_names = [name for name in names]
idx = 1
for idx, x_2 in enumerate(xxs, 2):
original_type = type(d1.get_data())
d2 = lgb.Dataset(x_2, feature_name=names, free_raw_data=False).construct()
......@@ -223,7 +222,6 @@ class TestBasic(unittest.TestCase):
self.assertIsInstance(d1.get_data(), original_type)
self.assertTupleEqual(d1.get_data().shape, (n_row, n_col * idx))
res_feature_names += ['D{}_{}'.format(idx, name) for name in names]
idx += 1
self.assertListEqual(d1.feature_name, res_feature_names)
def test_cegb_affects_behavior(self):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment