Updated script that can be controled by Nodejs web app
This commit is contained in:
@ -0,0 +1,40 @@
|
||||
import numpy as np
|
||||
import pytest
|
||||
|
||||
from pandas import (
|
||||
MultiIndex,
|
||||
Series,
|
||||
)
|
||||
import pandas._testing as tm
|
||||
|
||||
|
||||
class TestRepeat:
|
||||
def test_repeat(self):
|
||||
ser = Series(np.random.default_rng(2).standard_normal(3), index=["a", "b", "c"])
|
||||
|
||||
reps = ser.repeat(5)
|
||||
exp = Series(ser.values.repeat(5), index=ser.index.values.repeat(5))
|
||||
tm.assert_series_equal(reps, exp)
|
||||
|
||||
to_rep = [2, 3, 4]
|
||||
reps = ser.repeat(to_rep)
|
||||
exp = Series(ser.values.repeat(to_rep), index=ser.index.values.repeat(to_rep))
|
||||
tm.assert_series_equal(reps, exp)
|
||||
|
||||
def test_numpy_repeat(self):
|
||||
ser = Series(np.arange(3), name="x")
|
||||
expected = Series(
|
||||
ser.values.repeat(2), name="x", index=ser.index.values.repeat(2)
|
||||
)
|
||||
tm.assert_series_equal(np.repeat(ser, 2), expected)
|
||||
|
||||
msg = "the 'axis' parameter is not supported"
|
||||
with pytest.raises(ValueError, match=msg):
|
||||
np.repeat(ser, 2, axis=0)
|
||||
|
||||
def test_repeat_with_multiindex(self):
|
||||
# GH#9361, fixed by GH#7891
|
||||
m_idx = MultiIndex.from_tuples([(1, 2), (3, 4), (5, 6), (7, 8)])
|
||||
data = ["a", "b", "c", "d"]
|
||||
m_df = Series(data, index=m_idx)
|
||||
assert m_df.repeat(3).shape == (3 * len(data),)
|
Reference in New Issue
Block a user