def test_get_trade_history():
df = Instrument(1).get_trade_history(2)
assert df.to_csv(line_terminator='\n') == (
'date,pmax,pmin,pc,pl,pf,py,tval,tvol,tno\n'
'2021-01-20,10400.0,10120.0,10380.0,10400.0,10350.0,9910.0,498484813880.0,48013394,7284\n'
'2021-01-19,10380.0,9400.0,9910.0,10290.0,9400.0,9890.0,2649416188110.0,267389256,36765\n'
)
assert isinstance(df.index, DatetimeIndex)
def test_get_holders_without_cisin():
inst = Instrument(1)
assert inst.get_identification().loc['کد 12 رقمی شرکت',
1] == 'IRO7SDIP0002'
with patch.object(Instrument,
'get_identification',
side_effect=NotImplementedError) as get_identification:
with raises(NotImplementedError):
inst.get_holders()
get_identification.assert_called_once()
def test_get_holders_with_cisin():
holders = Instrument(1).get_holders(cisin=1)
assert holders.to_csv(line_terminator='\n') == (
',سهامدار/دارنده,سهم,درصد,تغییر,id_cisin\n'
'0,ETFکدرزروصندوقهای سرمایه گذاری قابل معامله,98 M,19.51,12 M,"21790,IRT3AVAF0003"\n'
'1,BFMصندوق سرمایه گذاری.ا.بازارگردانی معیار,24 M,4.82,1 M,"60500,IRT3AVAF0003"\n'
'2,صندوق سرمایه گذاری توازن معیار,10 M,2.05,0,"62783,IRT3AVAF0003"\n'
'3,شرکت توسعه سامانه تحلیل گری سپیدار-سهامی خاص-,9 M,1.83,-953440,"69558,IRT3AVAF0003"\n'
'4,شرکت مشاورسرمایه گذاری معیار-سهامی خاص-,9 M,1.82,0,"60679,IRT3AVAF0003"\n'
'5,شخص حقیقی,7 M,1.31,-10000,"69867,IRT3AVAF0003"\n'
'6,شرکت آرمان اندیشان رستاک-سهامی خاص-,5 M,1.0,0,"21346,IRT3AVAF0003"\n'
)
def test_get_page_info_no_eps():
d = Instrument(1).get_page_data()
assert d == {
'bvol': 1000000,
'cisin': 'IRO1KALA0001',
'cs': 67,
'eps': None,
'flow': 1,
'flow_name': 'بازار اول (تابلوی فرعی) بورس',
'free_float': 100,
'group_code': 'N1',
'isin': 'IRO1KALA0001',
'l18': 'کالا',
'l30': 'بورس کالای ایران',
'month_average_volume': 3575913,
'sector_name': 'فعالیتهای کمکی به نهادهای مالی واسط',
'sector_pe': 12.72,
'sps': None,
'tmax': 34410.0,
'tmin': 31150.0,
'week_max': 34900.0,
'week_min': 31150.0,
'year_max': 82400.0,
'year_min': 26430.0,
'z': 2500000000
}
def test_asam_instant():
assert Instrument(1).get_info(orders=True) == {
'status': 'A ',
'last_info_datetime': datetime(2020, 11, 11, 12, 28, 17),
'nav': 95630,
'nav_datetime': jdatetime(1399, 8, 21, 15, 13, 43),
'pc': 94140,
'pd1': 94000,
'pd2': 92350,
'pd3': 91140,
'pf': 94440,
'pl': 95890,
'pmax': 92001,
'pmin': 96000,
'po1': 95900,
'po2': 95990,
'po3': 96000,
'py': 93414,
'qd1': 5000,
'qd2': 50,
'qd3': 550,
'qo1': 3000,
'qo2': 250,
'qo3': 560,
'timestamp': '12:28:17',
'tno': 27,
'tval': 753116350,
'tvol': 8000,
'zd1': 1,
'zd2': 1,
'zd3': 1,
'zo1': 1,
'zo2': 2,
'zo3': 2
}
def test_get_adjustments():
adj_df = Instrument(1).get_adjustments()
assert adj_df.columns.tolist() == ['date', 'adj_pc', 'pc']
assert len(adj_df) == 18
assert adj_df.loc[0].values.tolist() == [
jdatetime(1399, 5, 1, 0, 0), 35720, 35970
]
def test_get_page_info_negative_sector_pe():
d = Instrument(1).get_page_data()
assert d == {
'bvol': 15842055,
'cisin': 'IRO1GOST0003',
'cs': 34,
'eps': 396,
'flow': 1,
'flow_name': 'بازار اول (تابلوی اصلی) بورس',
'free_float': 53,
'group_code': 'N1',
'isin': 'IRO1GOST0001',
'l18': 'خگستر',
'l30': 'گسترش\u200cسرمایه\u200cگذاری\u200cایران\u200cخودرو',
'month_average_volume': 366858421,
'sector_name': 'خودرو و ساخت قطعات',
'sector_pe': -9.7,
'sps': None,
'tmax': 2623.0,
'tmin': 2375.0,
'week_max': 2623.0,
'week_min': 2420.0,
'year_max': 12930.0,
'year_min': 1750.0,
'z': 39605137000
}
def test_get_page_info_no_free_float():
d = Instrument(1).get_page_data()
assert d == {
'bvol': 54000000,
'cisin': 'IRO3BDYZ0003',
'cs': 57,
'eps': 242,
'flow': 2,
'flow_name': 'بازار دوم فرابورس',
'free_float': None,
'group_code': 'Z1',
'isin': 'IRO3BDYZ0001',
'l18': 'دی',
'l30': 'بانک دی',
'month_average_volume': 0,
'sector_name': 'بانکها و موسسات اعتباری',
'sector_pe': 21.04,
'sps': None,
'tmax': 970.0,
'tmin': 878.0,
'week_max': 0.0,
'week_min': 0.0,
'year_max': 83000.0,
'year_min': 0.0,
'z': 135000000000
}
def test_get_adjusted_price_history():
adj_df = Instrument(1).get_price_history()
assert adj_df.columns.tolist() == [
'pmax', 'pmin', 'pf', 'pl', 'tvol', 'pc'
]
assert adj_df.index.name == 'date'
assert len(adj_df) == 18
assert adj_df.iloc[-1].values.tolist() == [
68410, 62366, 63500, 67508, 14222269, 65636
]
def test_get_client_type():
assert Instrument(1).get_client_type().to_csv(line_terminator='\n') == (
'date,n_buy_count,l_buy_count,n_sell_count,l_sell_count,n_buy_volume,l_buy_volume,n_sell_volume,l_sell_volume,n_buy_value,l_buy_value,n_sell_value,l_sell_value'
'\n2021-03-03,45996,40,536073,152,371944945,124311191,445956261,50299875,6032526703800,2001293803740,7203106931130,830713576410'
'\n2021-03-02,1390,5,68895,24,42463945,909997,29432662,13941280,664560739250,14241453050,460621160300,218181032000'
'\n2021-03-01,11957,10,208139,81,122577447,33815772,145381224,11011995,1810312954820,499453950210,2147119750880,162647154150'
'\n2021-02-28,32473,34,388441,374,300085412,16087286,185678823,130493875,4182394884300,224209104120,2587603120610,1819000867810'
'\n2021-02-27,1522,9,31838,20,10893007,80000,10965618,7389,143351972120,1052800000,144307532880,97239240'
'\n2021-02-24,2752985,1535,0,2,1014549773,480840379,0,1495390152,12645360378170,5928026392500,0,18573386770670'
'\n')
def test_dara1_instant():
assert Instrument(1).get_info(index=True, orders=True) == {
'derivatives_status': 'F',
'derivatives_tno': 7864,
'derivatives_tval': 150982456491.0,
'derivatives_tvol': 452251494.0,
'last_info_datetime': datetime(2021, 1, 27, 12, 30),
'market_last_transaction': jdatetime(1399, 11, 8, 15, 21, 59),
'nav': 190671,
'nav_datetime': jdatetime(1399, 11, 8, 15, 40),
'otc_status': 'F',
'otc_tno': 494135,
'otc_tval': 122240030535934.0,
'otc_tvol': 2168547032.0,
'pc': 151580,
'pd1': 150120,
'pd2': 150000,
'pd3': 149990,
'pf': 147550,
'pl': 150120,
'pmax': 141100,
'pmin': 158000,
'po1': 150120,
'po2': 150130,
'po3': 150500,
'py': 152030,
'qd1': 2000,
'qd2': 62729,
'qd3': 3185,
'qo1': 7275,
'qo2': 34582,
'qo3': 3862,
'status': 'A ',
'timestamp': '12:30:00',
'tno': 84083,
'tse_index': 1207698.27,
'tse_index_change': -7335.16,
'tse_index_change_percent': -0.6,
'tse_status': 'F',
'tse_tno': 113961561691999.0,
'tse_tval': 113961561691999.0,
'tse_tvol': 10062531582.0,
'tse_value': 4.802681498014614e+16,
'tval': 9972065145080,
'tvol': 65786166,
'zd1': 1,
'zd2': 8,
'zd3': 12,
'zo1': 3,
'zo2': 1,
'zo3': 3
}
def test_vsadid():
assert Instrument(1).get_info() == {
'status': 'IS',
'last_info_datetime': datetime(2021, 1, 24, 6, 41, 10),
'pc': 23810,
'pf': 0,
'pl': 23633,
'pmax': 0,
'pmin': 0,
'py': 23810,
'timestamp': '06:41:10',
'tno': 0,
'tval': 0,
'tvol': 0
}
def test_fmelli_instant():
assert Instrument(1).get_info(orders=False) == {
'status': 'A ',
'last_info_datetime': datetime(2020, 11, 11, 17, 29, 53),
'pc': 19890,
'pf': 19400,
'pl': 20320,
'pmax': 19200,
'pmin': 20320,
'py': 19360,
'timestamp': '17:29:53',
'tno': 26572,
'tval': 1318025925250,
'tvol': 66266936
}
def test_negin_instant():
assert Instrument(1).get_info() == {
'status': 'A ',
'last_info_datetime': datetime(2020, 11, 11, 12, 29, 37),
'nav': 12190,
'nav_datetime': jdatetime(1398, 12, 29, 16, 0),
'pc': 50110,
'pf': 50010,
'pl': 50000,
'pmax': 50000,
'pmin': 50700,
'py': 50010,
'timestamp': '12:29:37',
'tno': 29,
'tval': 1408671580,
'tvol': 28109
}
def test_get_holder():
inst = Instrument(1)
# has no other holdings
hist, oth = inst.get_holder('69867,IRT3AVAF0003', True, True)
assert hist.to_csv(line_terminator='\n').startswith(
'date,shares\n2021-03-01,6600001\n2021-03-02,6603001\n')
assert oth.to_csv(line_terminator='\n') == 'ins_code,name,shares,percent\n'
hist = inst.get_holder('69867,IRT3AVAF0003', True)
assert type(hist) is DataFrame
oth = inst.get_holder('69867,IRT3AVAF0003', False, True)
assert type(oth) is DataFrame
result = inst.get_holder('69867,IRT3AVAF0003', False)
assert oth.equals(result)
def test_get_identification():
assert Instrument(1).get_identification().to_dict() == {
1: {
'بازار': 'بازار پایه نارنجی فرابورس',
'زیر گروه صنعت': 'استخراج سایر فلزات اساسی',
'نام شرکت': 'گروه\u200cصنعتی\u200cسدید',
'نام لاتین شرکت': 'Sadid Group',
'نماد 30 رقمی فارسی': 'گروه \u200cصنعتی\u200cسدید',
'نماد فارسی': 'وسدید - لغو پذیرش شده',
'کد 12 رقمی شرکت': 'IRO7SDIP0002',
'کد 12 رقمی نماد': 'IRO7SDIP0001',
'کد 4 رقمی شرکت': 'SDIP',
'کد 5 رقمی نماد': 'SDIP1',
'کد تابلو': '7',
'کد زیر گروه صنعت': '2799',
'کد گروه صنعت': '27',
'گروه صنعت': 'فلزات اساسی'
}
}
def test_get_info_url(mock):
with raises(ValueError):
Instrument(1).get_info()
mock.assert_called_once_with(
'http://www.tsetmc.com/tsev2/data/instinfodata.aspx?i=1&c=&e=1')
def test_get_intraday_general():
result = Instrument(1).get_intraday(20210602,
general=True,
thresholds=True,
closings=True,
candles=True,
holders=True,
yesterday_holders=True)
assert len(result) == 10
assert result['general'] == {
'l30': 'ملی\u200c صنایع\u200c مس\u200c ایران\u200c',
'l18': 'فملی',
'market': 'بازار بورس',
'flow_name': 'بازار اول (تابلوی اصلی) بورس',
'flow': 1,
'group_code': 'N1',
'cisin': 'IRO1MSMI0000',
'isin': 'IRO1MSMI0001',
'z': 200000000000,
'bvol': 10143702
}
assert result['thresholds'].to_csv(
line_terminator='\n') == (',time,tmax,tmin\n'
'0,1,13080.0,11840.0\n'
'1,63122,13000.0,11780.0\n')
assert result['closings'].iloc[-1].to_dict() == {
'date': '1400/3/12 13:08:41',
'?1': '-',
'pl': '12480',
'pc': '12460',
'pf': '12190',
'py': '12390',
'pmin': '12590',
'pmax': '12190',
'tno': '10219',
'tvol': '60394598',
'tval': '752545395210',
'?2': '0',
'heven': '130841'
}
assert result['candles'].iloc[-1].to_dict() == {
'time': '12:15',
'high': 12580,
'low': 12400,
'open': 12440,
'close': 12460,
'tvol': 7036930
}
assert result['states'].iloc[-1].to_dict() == {
'date': 20210214,
'time': 1,
'state': 'A '
}
assert result['trades'].iloc[-1].to_dict() == {
'time': '12:30:00',
'tvol': 750,
'pl': 12480,
'annulled': False
}
assert len(result['holders']) == 0
yesterday_holders = result['yesterday_holders']
assert len(yesterday_holders) == 21
assert yesterday_holders.iloc[-1].to_dict() == {
'id': 21630,
'cisin': 'IRO1MSMI0000',
'shares': 2003857980,
'percent': 1.0,
'change': '',
'name': 'شرکت گروه توسعه مالی مهرآیندگان-سهامی عام-'
}
assert result['client_types'] == {
'l_buy_count': 10,
'l_buy_percent': 13,
'l_buy_value': 97592925820,
'l_buy_volume': 7835930,
'l_mean_buy_price': 12454.542832822652,
'l_mean_sell_price': 12460.894315227768,
'l_sell_count': 15,
'l_sell_percent': 19,
'l_sell_value': 140726287530,
'l_sell_volume': 11293434,
'n_buy_count': 4389,
'n_buy_percent': 87,
'n_buy_value': 654952469390,
'n_buy_volume': 52558668,
'n_mean_buy_price': 12461.359739748352,
'n_mean_sell_price': 12460.378896109265,
'n_sell_count': 0,
'n_sell_percent': 81,
'n_sell_value': 611819107680,
'n_sell_volume': 49101164
}
assert result['best_limits'].loc[133707].to_dict() == {
'row': 5,
'zd': 31,
'qd': 534434,
'pd': 12430,
'po': 12520,
'qo': 111906,
'zo': 15
}
def test_get_page_info():
d = Instrument(1).get_page_data(True, True, True)
trade_history = d.pop('trade_history')
related_companies = d.pop('related_companies')
assert d == {
'bvol': 9803922,
'cisin': 'IRO1MSMI0000',
'cs': 27,
'eps': 1339,
'flow': 1,
'flow_name': 'بازار اول (تابلوی اصلی) بورس',
'free_float': 33,
'group_code': 'N1',
'isin': 'IRO1MSMI0001',
'l18': 'فملی',
'l30': 'ملی\u200c صنایع\u200c مس\u200c ایران\u200c',
'month_average_volume': 80107878,
'sector_name': 'فلزات اساسی',
'sector_pe': 9.05,
'sps': 2452.07,
'tmax': 12850.0,
'tmin': 11630.0,
'week_max': 12440.0,
'week_min': 11990.0,
'year_max': 39810.0,
'year_min': 9400.0,
'z': 200000000000
}
assert related_companies == [
[
35425587644337450, 'فملی',
'ملی\u200c صنایع\u200c مس\u200c ایران\u200c'
], [46348559193224090, 'فولاد', 'فولاد مبارکه اصفهان'],
[9211775239375291, 'ذوب', 'سهامی ذوب آهن اصفهان'],
[28864540805361867, 'فخوز', 'فولاد خوزستان'],
[29974853866926823, 'فروی', 'ذوب روی اصفهان'],
[14800142337291217, 'فولای', 'صنایع فولاد آلیاژی یزد'],
[66701874099226162, 'فاسمین', 'کالسیمین\u200c'],
[43716452378323683, 'فزرین', 'زرین معدن آسیا'],
[41302553376174581, 'فجر', 'فولاد امیرکبیرکاشان'],
[43545527030854340, 'کویر', 'تولیدی فولاد سپید فراب کویر'],
[60350996279289099, 'کاوه', 'فولاد کاوه جنوب کیش'],
[
54277068923045214, 'فسرب',
'ملی\u200c سرب\u200cوروی\u200c ایران\u200c'
], [66772024744156373, 'فباهنر', 'مس\u200c شهیدباهنر'],
[408934423224097, 'فرآور', 'فرآوری\u200cموادمعدنی\u200cایران\u200c'],
[67170215467608124, 'زنگان', 'صنعت روی زنگان'],
[27814844870305607, 'کیمیا', 'معدنی کیمیای زنجان گستران'],
[59266699437480384, 'ارفع', 'شرکت آهن و فولاد ارفع'],
[18004480270695404, 'فمراد', 'آلومراد'],
[12874072841236826, 'فسازان', 'غلتک سازان سپاهان'],
[8977441217024425, 'فسپا', 'گروه\u200cصنعتی\u200cسپاهان\u200c'],
[24018878640527909, 'میدکو', 'هلدینگ صنایع معدنی خاورمیانه'],
[
48623320733330408, 'فلوله',
'لوله\u200cوماشین\u200cسازی\u200cایران\u200c'
], [70498485598181604, 'هرمز', 'فولاد هرمزگان جنوب'],
[
47232550823972469, 'وتوکا',
'سرمایه\u200cگذاری\u200cتوکافولاد(هلدینگ'
], [40808043719554948, 'فولاژ', 'فولاد آلیاژی ایران'],
[54419429862704331, 'فروس', 'فروسیلیس\u200c ایران\u200c'],
[44296315953738727, 'فاهواز', 'نورد و لوله اهواز'],
[4733285133017464, 'فخاس', 'فولاد خراسان'],
[65004959184388996, 'فایرا', 'آلومینیوم\u200cایران\u200c'],
[57875847776839336, 'فنوال', 'نورد آلومینیوم\u200c'],
[68488673556087148, 'فپنتا', 'سپنتا'],
[19367527798307032, 'فالوم', 'آلومتک\u200c'],
[41713045190742691, 'وسدید', 'گروه \u200cصنعتی\u200cسدید'],
[56324206651661881, 'فنورد', 'نوردوقطعات\u200c فولادی\u200c'],
[
20966291817819448, 'فسدید',
'لوله\u200cوتجهیزات\u200c سدید - ورشکسته'
], [35445515321658835, 'فماک', 'ماداکتو استیل کرد'],
[59342912854668427, 'فنفت', 'صنایع تجهیزات نفت'],
[66021783818850713, 'فافزا', 'فولاد افزا سپاهان'],
[66514709341259550, 'فوکا', 'فولاد کاویان'],
[55373808401388162, 'فتوسا', 'تولید و توسعه سرب روی ایرانیان'],
[8725363201030474, 'فگستر', 'گسترش صنایع روی ایرانیان'],
[41867071915439180, 'آلومینیوم جنوب', 'مجتمع آلومینیوم جنوب'],
[26881786652328215, 'زرند', 'فولاد زرند'],
[58903026391426893, 'سیرجان', 'فولاد سیرجان'],
[67535111875054076, 'پارس متال ', 'پارس متال'],
[42171673792069321, 'زیسکو', 'فولاد زرند ایرانیان'],
[40012411719639360, 'سیسکو', 'فولاد سیرجان ایرانیان'],
[48175603054578540, 'اسفراین', 'مجتمع صنعتی اسفراین'],
[357086043812735, 'فولاد تربت', 'فولاد تربت حیدریه'],
[68604686987554533, 'فماکح', 'ح . ماداکتو استیل کرد'],
[44846320603450383, 'فروسیل', 'فروسیلیسیم خمین'],
[37284308569715577, 'فسبزوار', 'پارس فولاد سبزوار'],
[45507655586782998, 'فجهان', 'مجتمع جهان فولاد سیرجان']
]
assert trade_history.to_csv(line_terminator='\n') == (
'date,pc,py,pmin,pmax,tno,tvol,tval\n'
'2021-06-30,12240.0,12240.0,12180.0,12370.0,11639,61924440,758028561610.0\n'
'2021-06-29,12240.0,12140.0,12110.0,12410.0,13153,80738158,988626346660.0\n'
'2021-06-28,12140.0,12220.0,11990.0,12290.0,12556,69479692,843417628060.0\n'
'2021-06-27,12220.0,12420.0,12040.0,12440.0,18830,93937722,1148372777010.0\n'
'2021-06-26,12420.0,12310.0,12120.0,12600.0,25260,155751582,1934122998780.0\n'
'2021-06-23,12310.0,11830.0,12020.0,12420.0,23635,204263514,2514120310650.0\n'
'2021-06-22,11830.0,11540.0,11530.0,12110.0,24234,170353210,2014436805300.0\n'
'2021-06-21,11540.0,11600.0,11290.0,11740.0,14337,71521163,825245262840.0\n'
'2021-06-20,11600.0,11880.0,11450.0,11960.0,14830,73966678,858202905300.0\n'
)
def test_equal():
assert Instrument.from_l18('فملی') == Instrument(35425587644337450)
def test_repr():
# known ID
assert repr(Instrument.from_l18('فملی')) == "Instrument('فملی')"
# unknown ID
assert repr(Instrument(1)) == "Instrument(1)"
def test_from_search_with_numeric_description():
# note the "30" in فيروزه - صندوق شاخص30 شركت فيروزه- سهام
assert Instrument.from_search('فیروزه').ins_code == 66036975502302203
def test_vskhooz_long():
assert Instrument(1).get_info() == VSKHOOZ
